python - 将字符串分成单独的列 pandas python

Question

我见过类似的问题，但没有真正符合我的问题。如果我有一个值表，例如:

value
a
b
b
c

我想使用 pandas 将列添加到表格中以显示例如:

value a b
a 1 0 
b 0 1
c 0 0

我尝试过以下方法:

df['a'] = 0

def string_count(indicator):
    if indicator == 'a':
        df['a'] == 1

df['a'].apply(string_count)

但这会产生:

 0    None
 1    None
 2    None
 3    None

我想至少达到选择被硬编码的程度(即我已经知道 a、b 和 c 出现)，但如果我可以设置字符串列，然后插入一个，那就更好了每个唯一字符串的列。

我是否以错误的方式处理这个问题？

Answer 1

dummies = pd.get_dummies(df.value)

   a  b  c
0  1  0  0
1  0  1  0
2  0  1  0
3  0  0  1

如果您只想显示唯一的出现次数，您可以添加:

dummies.index = df.value
dummies.drop_duplicates()

       a  b  c
value         
a      1  0  0
b      0  1  0
c      0  0  1

或者:

df = df.join(pd.get_dummies(df.value))

  value  a  b  c
0     a  1  0  0
1     b  0  1  0
2     b  0  1  0
3     c  0  0  1

您可以再次.drop_duplicates()仅查看值列中的唯一条目。