How do I convert text into multiple columns in python?

Question

Suppose now I have the following pandas data frame:

id text
1  A B C
2  B D
3  A D

And I want to get the following result:

I don't how to describe this transformation, it looks like one-hot encoding but they should be totally different.

Anyone knows how to do this transformation and what's the name of such transformation?

BENY · Accepted Answer · 2018-03-07 03:03:19Z

3

Something like str.get_dummies

pd.concat([df['id'],df.text.str.get_dummies(sep=' ')],1)
Out[249]: 
   id  A  B  C  D
0   1  1  1  1  0
1   2  0  1  0  1
2   3  1  0  0  1

answered Mar 7, 2018 at 3:03

BENY

324k22 gold badges176 silver badges250 bronze badges

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

2 Comments

Allen Qin Over a year ago

Nice use of get_dummies!

jpp Over a year ago

This version is better than mine. +1. My version is ignorant of sep parameter.

jpp · Accepted Answer · 2018-03-07 02:58:32Z

1

One way is via pd.get_dummies:

df = pd.DataFrame({'id': [1, 2, 3],
                   'text': ['A B C', 'B D', 'A D']})

df['text'] = df['text'].str.split(' ').str.join('|')

df = df.join(df['text'].str.get_dummies()).drop('text', 1)

#    id  A  B  C  D
# 0   1  1  1  1  0
# 1   2  0  1  0  1
# 2   3  1  0  0  1

answered Mar 7, 2018 at 2:58

jpp

166k37 gold badges301 silver badges362 bronze badges

Collectives™ on Stack Overflow

How do I convert text into multiple columns in python?

2 Answers 2

2 Comments

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

2 Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Related