如何旋轉pandas DataFrame列以創建二進制“值表”？

Question

我有以下熊貓數據框：

import pandas as pd
df = pd.read_csv("filename.csv")

df 
     A   B         C         D        E    
0    a  0.469112 -0.282863 -1.509059  cat  
1    c -1.135632  1.212112 -0.173215  dog   
2    e  0.119209 -1.044236 -0.861849  dog   
3    f -2.104569 -0.494929  1.071804  bird   
4    g -2.224569 -0.724929  2.234213  elephant
...

我想基於column E的分類值的標識創建更多列，以使數據框如下所示：

 df 
         A   B         C         D        cat    dog     bird    elephant ....    
    0    a  0.469112 -0.282863 -1.509059  -1      0       0       0
    1    c -1.135632  1.212112 -0.173215   0     -1       0       0
    2    e  0.119209 -1.044236 -0.861849   0     -1       0       0
    3    f -2.104569 -0.494929  1.071804   0      0      -1       0
    4    g -2.224569 -0.724929  2.234213   0      0       0       0
    ...

也就是說，我轉動的值列E是基於對值的二進制矩陣E ，給1 ，如果該值存在，並且0的地方沒有其他所有（在這里，我想它是-1或“負二進制矩陣”）？

我不確定pandas中哪個函數最能做到這一點：也許pandas.DataFrame.unstack()嗎？

任何見解表示贊賞！

Answer 1

使用pd.concat ， drop和get_dummies

pd.concat([df.drop('E', 1), pd.get_dummies(df.E).mul(-1)], axis=1)