python pandas:将逗号分隔的列拆分为新列-每个值一个 [英] python pandas: split comma-separated column into new columns - one per value

问题描述

我有一个像这样的数据框:

I have a dataframe like this:

data = np.array([["userA","event2, event3"],
            ['userB',"event3, event4"],
            ['userC',"event2"]])

data = pd.DataFrame(data)

        0         1
0   userA   "event2, event3"
1   userB   "event3, event4"
2   userC   "event2"

现在我想获得一个这样的数据框:

now I would like to get a dataframe like this:

       0    event2      event3      event4
0   userA     1           1
1   userB                 1           1
2   userC     1

有人可以帮忙吗?

推荐答案

似乎您需要

It seems you need get_dummies with replace 0 to empty strings:

df = data[[0]].join(data[1].str.get_dummies(', ').replace(0, ''))
print (df)
       0 event2 event3 event4
0  userA      1      1       
1  userB             1      1
2  userC      1              

详细信息:

print (data[1].str.get_dummies(', '))
   event2  event3  event4
0       1       1       0
1       0       1       1
2       1       0       0

这篇关于python pandas:将逗号分隔的列拆分为新列-每个值一个的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！