Skip to content

BUG: Using categorical dtype in read_json #25573

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Closed
wants to merge 2 commits into from
Closed

BUG: Using categorical dtype in read_json #25573

Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
fdcea63
Using categorical dtype in read_json
anki-code Mar 6, 2019
e3bb759
PEP8 fix
anki-code Mar 6, 2019
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
6 changes: 5 additions & 1 deletion pandas/io/json/json.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,7 @@
from pandas.compat import StringIO, long, to_str, u
from pandas.errors import AbstractMethodError

from pandas.core.dtypes.common import is_period_dtype
from pandas.core.dtypes.common import is_period_dtype, is_categorical_dtype

from pandas import DataFrame, MultiIndex, Series, compat, isna, to_datetime
from pandas.core.reshape.concat import concat
Expand Down Expand Up @@ -712,6 +712,10 @@ def _try_convert_data(self, name, data, use_dtypes=True,
# dtype to force
dtype = (self.dtype.get(name)
if isinstance(self.dtype, dict) else self.dtype)

if is_categorical_dtype(dtype):
return data.astype('category'), True

if dtype is not None:
try:
dtype = np.dtype(dtype)
Expand Down