如何在复杂的条件下获得假人

Question

我有下面的数据框

   col  
A  a
A  b
A  c
B  d
B  e
C  f

我想要获取虚拟变量

   a b c d e f 
A  1 1 1 0 0 0 
B  0 0 0 1 1 0  
C  0 0 0 0 0 1

我怎么才能得到这个呢？

我试过了

df.col.get_dummies()

但我不能组队。

Answer 1

您需要按索引和聚合max的groupby

print (df.col.str.get_dummies().groupby(level=0).max())
   a  b  c  d  e  f
A  1  1  1  0  0  0
B  0  0  0  1  1  0
C  0  0  0  0  0  1

或者：

print (pd.get_dummies(df.col).groupby(level=0).max())
   a  b  c  d  e  f
A  1  1  1  0  0  0
B  0  0  0  1  1  0
C  0  0  0  0  0  1