编程语言
首页 > 编程语言> > python-如何透视pandas DataFrame列以创建二进制“值表”?

python-如何透视pandas DataFrame列以创建二进制“值表”?

作者:互联网

我有以下熊猫数据框:

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
...

我想基于列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()?

任何见解表示赞赏!

解决方法:

使用pd.concat,drop和get_dummies

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

enter image description here

标签:categorical-data,pandas,dataframe,binary,python
来源: https://codeday.me/bug/20191026/1938055.html