python-2.7 - Pandas / python : split column based on a dictionary

Question

我有一个 Pandas 数据框，其中有一列包含逗号分隔值:

a,c,f
a,b,c,d,e
c,g,f,b
..

每一行都有不同的长度、不同的顺序和值。这是 a 到 m 的字母。

我想用 0 和 1 将它转换成 13 列:

a b c d e f g h i j k l m
1 0 0 0 0 1 0 0 0 0 0 0 0
1 1 1 1 1 0 0 0 0 0 0 0 0
0 1 1 1 1 0 0 0 0 0 0 0 0

我如何在 pandas/python 2.7 中有效地做到这一点？我考虑过根据列中是否存在子字符串的测试来创建一个列。但我不敢相信没有更有效的方法来做到这一点 :)

Answer 1

您可以使用 get_dummies如果列是 a:

print df.a
0        a,c,f
1    a,b,c,d,e
2      c,g,f,b
Name: a, dtype: object

print df.a.str.get_dummies(sep=',')
   a  b  c  d  e  f  g
0  1  0  1  0  0  1  0
1  1  1  1  1  1  0  0
2  0  1  1  0  0  1  1