File tree Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -474,16 +474,16 @@ def to_dummies(self, na_column=None) -> "DataFrame":
474
474
--------
475
475
:func:`pandas.get_dummies`
476
476
"""
477
- from pandas import DataFrame , CategoricalIndex
477
+ from pandas import DataFrame , CategoricalIndex , Series
478
478
479
479
eye = np .eye (len (self .categories ) + 1 , dtype = bool )
480
480
arr = eye [self .codes , :]
481
481
482
482
if na_column is None :
483
483
return DataFrame (arr [:, :- 1 ], columns = CategoricalIndex (self .categories ))
484
484
else :
485
- cat_lst = list (self .categories ) + [na_column ]
486
- return DataFrame (arr , columns = CategoricalIndex ( cat_lst ) )
485
+ cats = CategoricalIndex ( Series ( list (self .categories ) + [na_column ]))
486
+ return DataFrame (arr , columns = cats )
487
487
488
488
@property
489
489
def categories (self ):
You can’t perform that action at this time.
0 commit comments