如何在复杂条件下得到假人

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：

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