@@ -2422,23 +2422,21 @@ def get_dummies(
2422
2422
)
2423
2423
from pandas .core .reshape .encoding import get_dummies
2424
2424
2425
- input_series = Series (self ._data ) if isinstance (self ._data , ABCIndex ) else self ._data
2425
+ input_series = (
2426
+ Series (self ._data ) if isinstance (self ._data , ABCIndex ) else self ._data
2427
+ )
2426
2428
string_series = input_series .apply (lambda x : str (x ) if not isna (x ) else x )
2427
2429
split_series = string_series .str .split (sep , expand = True ).stack ()
2428
2430
valid_split_series = split_series [
2429
- (split_series .astype (str ) != 'None' ) &
2430
- ~ (split_series .index .get_level_values (0 ).duplicated (keep = 'first' ) & split_series .isna ())
2431
+ (split_series .astype (str ) != "None" )
2432
+ & ~ (
2433
+ split_series .index .get_level_values (0 ).duplicated (keep = "first" )
2434
+ & split_series .isna ()
2435
+ )
2431
2436
]
2432
2437
2433
2438
dummy_df = get_dummies (
2434
- valid_split_series ,
2435
- None ,
2436
- None ,
2437
- dummy_na ,
2438
- None ,
2439
- sparse ,
2440
- False ,
2441
- dtype
2439
+ valid_split_series , None , None , dummy_na , None , sparse , False , dtype
2442
2440
)
2443
2441
grouped_dummies = dummy_df .groupby (level = 0 )
2444
2442
if dtype == bool :
@@ -2447,7 +2445,9 @@ def get_dummies(
2447
2445
result_df = grouped_dummies .sum ()
2448
2446
2449
2447
if isinstance (prefix , str ):
2450
- result_df .columns = [f"{ prefix } { prefix_sep } { col } " for col in result_df .columns ]
2448
+ result_df .columns = [
2449
+ f"{ prefix } { prefix_sep } { col } " for col in result_df .columns
2450
+ ]
2451
2451
elif isinstance (prefix , dict ):
2452
2452
if len (prefix ) != len (result_df .columns ):
2453
2453
len_msg = (
@@ -2456,7 +2456,9 @@ def get_dummies(
2456
2456
f"({ len (result_df .columns )} )."
2457
2457
)
2458
2458
raise ValueError (len_msg )
2459
- result_df .columns = [f"{ prefix [col ]} { prefix_sep } { col } " for col in result_df .columns ]
2459
+ result_df .columns = [
2460
+ f"{ prefix [col ]} { prefix_sep } { col } " for col in result_df .columns
2461
+ ]
2460
2462
elif isinstance (prefix , list ):
2461
2463
if len (prefix ) != len (result_df .columns ):
2462
2464
len_msg = (
@@ -2465,7 +2467,10 @@ def get_dummies(
2465
2467
f"({ len (result_df .columns )} )."
2466
2468
)
2467
2469
raise ValueError (len_msg )
2468
- result_df .columns = [f"{ prefix [i ]} { prefix_sep } { col } " for i , col in enumerate (result_df .columns )]
2470
+ result_df .columns = [
2471
+ f"{ prefix [i ]} { prefix_sep } { col } "
2472
+ for i , col in enumerate (result_df .columns )
2473
+ ]
2469
2474
2470
2475
if isinstance (self ._data , ABCIndex ):
2471
2476
return MultiIndex .from_frame (result_df )
0 commit comments