列中具有多个标签的一种热编码

2022-01-21 00:00:00 python pandas dataset one-hot-encoding

问题描述

我有一个简单的数据集.

I have a simple dataset.

id,question,category,tags,day,quarter,group_id

1,What is your name,Introduction,Introduction,1,3,0

2,What is your name,Introduction,"Introduction, work",1,3,1

现在,如果您看到,在 tags 列中有多个用逗号分隔的输入.如果我尝试使用 pandas get_dummies 函数进行一次热编码,我会将其作为单列获取.但我想为每个标签创建列.我怎么可能做到这一点?

Now if you see, in the tags column there are multiple inputs seperated by commas. If I try to one-hot-encode using pandas get_dummies function I will get that as a single column. But I wanted to create columns for each tags. How can I do that possibly?


解决方案

相信需要str.get_dummies:

df1 = df['tags'].str.get_dummies(', ')
print (df1)

   Introduction  work
0             1     0
1             1     1

相关文章