Skip to content

Commit 35ec501

Browse files
committed
Explicit integer type cast
1 parent db173eb commit 35ec501

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

pandas/core/arrays/categorical.py

+2-3
Original file line numberDiff line numberDiff line change
@@ -457,9 +457,8 @@ def from_dummies(
457457
# 010 020 2 1
458458
# 001 * 1,2,3 => 003 -> 3 -> 2 = correct codes
459459
# 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)
460+
codes = ((df * mult_by).sum(axis=1) - 1).astype("Int64")
461+
return cls.from_codes(codes.fillna(-1), df.columns.values, ordered=ordered)
463462

464463
def to_dummies(self, na_column=None) -> "DataFrame":
465464
"""

0 commit comments

Comments
 (0)