pd.get_dummies() 来自带有整数0 和 1的列

编程问答 2022-05-25

问题描述

我有一个包含三列的数据框，一个是浮点/连续变量（高度），两个是整数/分类变量（糖尿病、性别）。

我想为分类变量生成虚拟变量，以便我可以构建回归模型。

问题是我在尝试使用 KeyError: ('diabetes','sex') 时收到错误消息 (pd.get_dummies())。

我不知道为什么会收到错误消息。我认为使用 .astype(str) 会解决这个问题，因为我过去从字符串变量生成哑元没有遇到过问题。

感谢您的帮助！

# library 
import pandas as pd

# data 
df = pd.DataFrame({'height': [1.77,1.54,1.52],'diabetes': [0,1],'sex': [1,1]})

# Failed attempt at generating dummies 
dummies = pd.get_dummies(df['diabetes','sex'].astype(str),prefix=['diabetes','sex'])


# expected outcome/goal  
expected = pd.DataFrame({'diabetes_0': [1,1,0],'diabetes_1': [0,'sex_0': [0,'sex_1': [1,1]})

解决方法

暂无找到可以解决该程序问题的有效方法，小编努力寻找整理中！

如果你已经找到好的解决方法，欢迎将解决方案带上本链接一起发送给小编。

小编邮箱:dio#foxmail.com (将#修改为@）

pandaspandaspythonregression