We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 576a9c6 commit a1ab12bCopy full SHA for a1ab12b
pandas/core/arrays/categorical.py
@@ -474,16 +474,16 @@ def to_dummies(self, na_column=None) -> "DataFrame":
474
--------
475
:func:`pandas.get_dummies`
476
"""
477
- from pandas import DataFrame, CategoricalIndex
+ from pandas import DataFrame, CategoricalIndex, Series
478
479
eye = np.eye(len(self.categories) + 1, dtype=bool)
480
arr = eye[self.codes, :]
481
482
if na_column is None:
483
return DataFrame(arr[:, :-1], columns=CategoricalIndex(self.categories))
484
else:
485
- cat_lst = list(self.categories) + [na_column]
486
- return DataFrame(arr, columns=CategoricalIndex(cat_lst))
+ cats = CategoricalIndex(Series(list(self.categories) + [na_column]))
+ return DataFrame(arr, columns=cats)
487
488
@property
489
def categories(self):
0 commit comments