Skip to content

Commit 717b888

Browse files
committed
PERF: do not expand categoricals when re-coding
1 parent deafc21 commit 717b888

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

pandas/core/categorical.py

+6
Original file line numberDiff line numberDiff line change
@@ -2219,6 +2219,12 @@ def _get_codes_for_values(values, categories):
22192219
"""
22202220

22212221
from pandas.core.algorithms import _get_data_algo, _hashtables
2222+
if is_categorical_dtype(values.dtype):
2223+
codes = (values.cat.codes if isinstance(values, ABCSeries)
2224+
else values.codes)
2225+
return _recode_for_categories(codes, values.dtype.categories,
2226+
categories)
2227+
22222228
if not is_dtype_equal(values.dtype, categories.dtype):
22232229
values = _ensure_object(values)
22242230
categories = _ensure_object(categories)

0 commit comments

Comments
 (0)