时间戳

首页 > 解决方案 > Pandas Python用户输入数据框对象的属性

python - Pandas Python用户输入数据框对象的属性

问题描述

我试图允许用户输入数据框对象的属性。

我尝试将输入更改为字符串。我也尝试过使用保存到变量的输入。这两个选项都不起作用。

data = pd.read_csv('2019FallEnrollees.csv')

input1_col = input("Enter comparison group A: ")
input2_col = input("Enter comparison group B: ")

input1_str= str(input1_col)
input2_str = str(input2_col)

test = data[['CUM_GPA', input1_str, input2_str]]

# error here! 'test' does not have attribute 'input1_str' or 'input1_col'
df_1 = test[(test.input1_str == 0) & (test.input2_str == 0)]
df_2 = test[(test.input1_col == 1) & (test.input2_col == 0)]

print(stats.ttest_ind(df_1.CUM_GPA, df_2.CUM_GPA, equal_var = False))

错误消息说

"AttributeError: 'DataFrame' 对象没有属性 'input1_str' 或 "AttributeError: 'DataFrame' 对象没有属性 'input1_col'

标签： pythonpandas

解决方案

欢迎！要访问 pandas 中的列，您不能使用data.column Trydata[column]或在您的情况下test[input1_col] 在这样做之前，请确保该列确实存在并且用户没有输入不存在的列。

有时列名可以是整数，转换为字符串也可能是一个问题

您可以通过运行获取所有数据框列的列表data.columns（如果您想要一个常规数组list(data.columns)：），实际上您可以通过运行更改列名data.columns = ["Column Header 1" , "Column Header 2" etc.]

推荐阅读