Skip to content

Commit 049472e

Browse files
committed
Explicit integer type cast
1 parent f8976ad commit 049472e

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

pandas/core/arrays/categorical.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -448,9 +448,8 @@ def from_dummies(
448448
# 010 020 2 1
449449
# 001 * 1,2,3 => 003 -> 3 -> 2 = correct codes
450450
# 100 100 1 0
451-
codes = (df * mult_by).sum(axis=1) - 1
452-
codes[codes.isna()] = -1
453-
return cls.from_codes(codes, df.columns.values, ordered=ordered)
451+
codes = ((df * mult_by).sum(axis=1) - 1).astype("Int64")
452+
return cls.from_codes(codes.fillna(-1), df.columns.values, ordered=ordered)
454453

455454
def to_dummies(self, na_column=None) -> "DataFrame":
456455
"""

0 commit comments

Comments
 (0)