Skip to content

Commit 7ccaf47

Browse files
committed
Properly handle NaN
1 parent d95a42b commit 7ccaf47

File tree

1 file changed

+3
-5
lines changed

1 file changed

+3
-5
lines changed

pandas/core/strings.py

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1001,13 +1001,11 @@ def str_get_dummies(arr, sep='|'):
10011001
2 1 0 1
10021002
"""
10031003
arr = arr.fillna('')
1004-
try:
1005-
arr = sep + arr + sep
1006-
except TypeError:
1007-
arr = sep + arr.astype(str) + sep
10081004

10091005
arr_split = arr.str.split(sep)
1010-
stacked = np.concatenate(arr_split)
1006+
1007+
stacked = Series(np.concatenate(arr_split))
1008+
stacked[stacked == ''] = np.nan
10111009
stacked_idx = np.repeat(np.arange(len(arr)), arr_split.str.len())
10121010

10131011
dummies_stacked = get_dummies(stacked)

0 commit comments

Comments
 (0)