Creating new binary columns from single string column in pandas

Question

I've seen this before and simply can't remember the function.

Say I have a column "Speed" and each row has 1 of these values:

'Slow', 'Normal', 'Fast'

How do I create a new dataframe with all my rows except the column "Speed" which is now 3 columns: "Slow" "Normal" and "Fast" which has all of my rows labeled with a 1 in whichever column the old "Speed" column was. So if I had:

print df['Speed'].ix[0]
> 'Normal'

I would not expect this:

print df['Normal'].ix[0]
>1

print df['Slow'].ix[0]
>0

joris · Accepted Answer · 2014-03-24 22:33:40Z

14

You can do this easily with pd.get_dummies (docs):

In [37]: df = pd.DataFrame(['Slow', 'Normal', 'Fast', 'Slow'], columns=['Speed'])

In [38]: df
Out[38]:
    Speed
0    Slow
1  Normal
2    Fast
3    Slow

In [39]: pd.get_dummies(df['Speed'])
Out[39]:
   Fast  Normal  Slow
0     0       0     1
1     0       1     0
2     1       0     0
3     0       0     1

answered Mar 24, 2014 at 22:33

joris

140k37 gold badges257 silver badges207 bronze badges

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

1 Comment

user1610719 Over a year ago

get_dummies was it! Thanks!

aha · Accepted Answer · 2014-03-24 22:27:19Z

6

Here is one solution:

df['Normal'] = df.Speed.apply(lambda x: 1 if x == "Normal" else 0)
df['Slow'] = df.Speed.apply(lambda x: 1 if x == "Slow" else 0)
df['Fast'] = df.Speed.apply(lambda x: 1 if x == "Fast" else 0)

answered Mar 24, 2014 at 22:27

aha

4,6744 gold badges26 silver badges27 bronze badges

Comments

sun · Accepted Answer · 2019-08-15 01:56:27Z

2

This has another method：

df           = pd.DataFrame(['Slow','Fast','Normal','Normal'],columns=['Speed'])
df['Normal'] = np.where(df['Speed'] == 'Normal', 1 ,0)
df['Fast']   = np.where(df['Speed'] == 'Fast', 1 ,0)
df['Slow']   = np.where(df['Speed'] == 'Slow', 1 ,0)

df 
     Speed  Normal  Fast  Slow
0    Slow       0     0     1
1    Fast       0     1     0
2  Normal       1     0     0
3  Normal       1     0     1

answered Aug 15, 2019 at 1:56

sun

1931 silver badge7 bronze badges

Collectives™ on Stack Overflow

Creating new binary columns from single string column in pandas

3 Answers 3

1 Comment

Comments

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

3 Answers 3

1 Comment

Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Related