How to convert column of DataFrame with n values to n binary columns?

Question

I am prety sure that there is a pandas function to do so but I could not find it when I was looking at the documentation and googled it. Maybe you can help me. Here is what I want to do:

import pandas as pd

data = {'col1': ['val1','val2','val3'],
        'col2': ['feat1','feat1','feat2']
       }
df = pd.DataFrame(data)
print (df)

>    col1   col2
> 0  val1  feat1
> 1  val2  feat1
> 2  val3  feat2

and I want to have the dataframe with this shape:

>    col1  feat1  feat2
> 0  val1      1      0
> 1  val2      1      0
> 2  val3      0      1

Best P

You need .get_dummies() method.

Always Right Never Left
– Always Right Never Left

2020-09-11 13:09:42 +00:00
Commented Sep 11, 2020 at 13:09 — Always Right Never Left
– Always Right Never Left, Commented Sep 11, 2020 at 13:09

Andrej Kesely · Accepted Answer · 2020-09-11 13:11:35Z

1

print(pd.get_dummies(df.set_index('col1'), prefix='', prefix_sep=''))

Prints:

      feat1  feat2
col1              
val1      1      0
val2      1      0
val3      0      1

answered Sep 11, 2020 at 13:11

Andrej Kesely

196k15 gold badges60 silver badges105 bronze badges

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

Comments

YOLO · Accepted Answer · 2020-09-11 13:19:00Z

0

You can also use crosstab

pd.crosstab(df.col1, df.col2)

answered Sep 11, 2020 at 13:19

YOLO

22k5 gold badges25 silver badges42 bronze badges

Collectives™ on Stack Overflow

How to convert column of DataFrame with n values to n binary columns?

2 Answers 2

Comments

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Related