We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent db173eb commit 35ec501Copy full SHA for 35ec501
pandas/core/arrays/categorical.py
@@ -457,9 +457,8 @@ def from_dummies(
457
# 010 020 2 1
458
# 001 * 1,2,3 => 003 -> 3 -> 2 = correct codes
459
# 100 100 1 0
460
- codes = (df * mult_by).sum(axis=1) - 1
461
- codes[codes.isna()] = -1
462
- return cls.from_codes(codes, df.columns.values, ordered=ordered)
+ codes = ((df * mult_by).sum(axis=1) - 1).astype("Int64")
+ return cls.from_codes(codes.fillna(-1), df.columns.values, ordered=ordered)
463
464
def to_dummies(self, na_column=None) -> "DataFrame":
465
"""
0 commit comments