使用pandas将字符串列表转换为虚拟变量

Question

如果我有以下熊猫DataFrame：

pd.DataFrame(columns=['name', 'tags'], data=[
    ['Rob', ['a', 'c']],
    ['Erica', ['b', 'c']]
])

表格：

^{pr2}$

如何将其转换为：

Name   tags_a  tags_b  tags_c
Rob    1       0       1
Erica  0       1       1

如果每一行只能有一个标记，我可以用pd.get_dummies(df, columns=['tags'])来做，但是当tags是List时，这就不起作用了。在

Answer 1

str.get_dummies

df.tags.str.join('|').str.get_dummies().add_prefix('tags_')

   tags_a  tags_b  tags_c
0       1       0       1
1       0       1       1

包括join

^{pr2}$