Create columns in dataframe based on csv field

Question

I have a pandas dataframe with the column "Values" that has comma separated values:

Row|Values
1|1,2,3,8
2|1,4

I want to create columns based on the CSV, and assign a boolean indicating if the row has that value, as follows:

Row|1,2,3,4,8
1|true,true,true,false,true
2|true,false,false,true,false

How can I accomplish that?

Thanks in advance

BENY · Accepted Answer · 2018-11-01 16:26:24Z

4

Just using get_dummies, check the link here and the astype(bool) change 1 to True 0 to False

df.set_index('Row')['Values'].str.get_dummies(',').astype(bool)
Out[318]: 
        1      2      3      4      8
Row                                  
1    True   True   True  False   True
2    True  False  False   True  False

edited Nov 1, 2018 at 16:26

answered Nov 1, 2018 at 16:22

BENY

324k22 gold badges176 silver badges250 bronze badges

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

1 Comment

BENY Over a year ago

@pygo Just the base usage for get_dummies function , I will add the link

Collectives™ on Stack Overflow

Create columns in dataframe based on csv field

1 Answer 1

1 Comment

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

1 Comment

Your Answer

Sign up or log in

Post as a guest

Related