动态过滤熊猫数据框时出现KeyError

Question

尝试过滤数据框中的列时，主题行出现以下错误（请参阅附件的屏幕截图以获取详细信息）：

  

[Index（[''a'，'c'']，dtype ='object'）]都不在[列]中

这是我的代码：

seleted_columns_final = "'a','c'"

df_final = df[[str(seleted_columns_final)]]
print(df_final)

测试数据帧如下：

row1 = {'a':5,'b':6,'c':7,'d':'A'}
row2 = {'a':8,'b':9,'c':10,'d':'B'}
row3 = {'a':11,'b':12,'c':13,'d':'C'}
df = pd.DataFrame([row1,row2,row3])[enter image description here][2]

Answer 1

您已将seleted_columns_final声明为字符串（“'a'，'c'”），然后再次将其转换为字符串？我猜您想将选定列的列表作为列表，并使用该列表创建数据框。

尝试：

seleted_columns_final= ['a','c']
df_final = df[seleted_columns_final]
print(df_final)

输出：

    a   c
0   5   7
1   8  10
2  11  13