python - 我如何从 pandas 数据帧创建二进制矩阵？

Question

我有一个看起来像这样的 df:

> df.head()

    Name     Application
   manager1     word
   manager2     mail
   manager2     internet
   manager1     excel
   manager3     word

我尝试从中创建一个二进制矩阵，但找不到解决方案。

Name      excel   internet  mail  word
manager1    1         0       0     1
manager2    0         1       1     0 
manager3    0         0       0     1 

Answer 1

使用get_dummies与最大:

df1 = pd.get_dummies(df.set_index('Name')['Application']).max(level=0).reset_index()
print (df1)
       Name  excel  internet  mail  word
0  manager1      1         0     0     1
1  manager2      0         1     1     0
2  manager3      0         0     0     1

或者使用聚合max:

df1 = (pd.get_dummies(df, columns=['Application'], prefix='', prefix_sep='')
         .groupby('Name', as_index=False)
         .max())
print (df1)
       Name  excel  internet  mail  word
0  manager1      1         0     0     1
1  manager2      0         1     1     0
2  manager3      0         0     0     1