Python – 将Pandas DataFrame转换为二进制数据

使用get_dummies（）方法将分类DataFrame转换为二进制数据。以下是我们的Pandas DataFrame，有2列 –

dataFrame = pd.DataFrame(
   {
      "学生": ['Jack', 'Robin', 'Ted', 'Scarlett', 'Kat'],"结果": ['通过', '失败', '失败', '通过', '通过']
   }
)

使用get_dummies（）并设置要转换为二进制形式的列。在这里，我们想要“通过”和“失败”形式的结果可见。因此，我们将设置“Result”列-

pd.get_dummies(dataFrame["结果"]

更多Pandas相关文章，请阅读：Pandas 教程

示例

以下是代码-

import pandas as pd

#创建DataFream
dataFrame = pd.DataFrame(
   {
      "学生": ['Jack', 'Robin', 'Ted', 'Scarlett', 'Kat'],"结果": ['通过', '失败', '失败', '通过', '通过']
   }
)

print"DataFrame ...\n",dataFrame

#转换为二进制数据
dfBinary = pd.get_dummies(dataFrame["结果"])
print"\n以二进制形式显示DataFrame ...\n",dfBinary

输出

这将产生以下输出 –

DataFrame ...
   结果   学生
0    通过    Jack
1    失败   Robin
2    失败      Ted
3    通过 Scarlett
4    通过      Kat

以二进制形式显示DataFrame ...
   失败    通过
0     0      1
1     1      0
2     1      0
3     0      1
4     0      1