标签二值化器：多列

Question

我目前正在尝试转换下面的数据框：

0                Cat             Chicken
1                Fish            Chicken
2                Fish            Cat
3                Cat             Chicken
4                Dog             Cat

到

                Cat    Fish   Dog    Chicken
0                1      0      0      1     
1                0      1      0      1
2                1      1      0      0
3                1      0      0      1
4                1      0      1      0

到目前为止，我只成功地完成了单列的标签二值化器。我可以想象以蛮力的方式执行此操作，循环遍历所有内容并有条件地检查。但是，这会导致运行时间很长。因此，有没有更聪明的方法来做到这一点？

Answer 1

使用带有前缀和前缀分隔符的get_dummies作为空字符串，即

pd.get_dummies(df,prefix='',prefix_sep='').groupby(level=0, axis=1).max() # Thanks @ coldspeed

     Cat  Chicken  Dog  Fish
0    1        1    0     0
1    0        1    0     1
2    1        0    0     1
3    1        1    0     0
4    1        0    1     0

Answer 2

df.apply(','.join,1).str.get_dummies(sep=',')
Out[51]: 
   Cat  Chicken  Dog  Fish
0    1        1    0     0
1    0        1    0     1
2    1        0    0     1
3    1        1    0     0
4    1        0    1     0