Code working not as expected in string splitting

Question

I have dataframe like this.

 df['category'] 
0    None
1    None
2    None
3    None
Name: category, dtype: object

I want applying the below code and splitting them to 2 columns as level1 and level2

Code

if df['category'] is None:
    riskc = pd.DataFrame(columns=['Level1', 'Level2'])
else:
    riskc=df['category'].str.split(',', expand=True).melt()['value']\
          .str.split(':', expand=True).rename(columns={0:'Level1', 1:'Level2'})

Current Output

riskc

    Level1
0   None
1   None
2   None
3   None

Expected Output

riskc

    Level1      Level2
0   None        None
0   None        None
1   None        None
2   None        None
3   None        None

What is wrong in my code

Allen Qin · Accepted Answer · 2021-05-31 03:26:53Z

1

To check if df['category'] contains all Nones, you need to use:

if df['category'].isna().all():
    riskc = pd.DataFrame(columns=['Level1', 'Level2'])

answered May 31, 2021 at 3:26

Allen Qin

20k9 gold badges55 silver badges68 bronze badges

Sign up to request clarification or add additional context in comments.

2 Comments

aeapen Over a year ago

Works, what was the mistake I made

Allen Qin Over a year ago

You are comparing df['category'] to None but df['category'](even when all of its elements are None) is a Series and not a None.

Collectives™ on Stack Overflow

Code working not as expected in string splitting

1 Answer 1

2 Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

2 Comments

Your Answer

Sign up or log in

Post as a guest

Related