如何通过字典列表值选择Pandas DataFrame列？解决类型错误

嘿，这个问题我之前也踩过坑！你遇到的TypeError: unhashable type: 'list'错误，根源是你给Pandas传递了嵌套列表作为列索引，而不是它能识别的列名列表。

问题出在哪？ #

你写的代码里：

df_out = df_in[[ds['cols']]]

这里的ds['cols']本身已经是一个列表['b', 'c']了，你再给它套一层方括号，就变成了[['b', 'c']]——也就是一个包含列表的列表。Pandas会把这个内层列表当成一个单独的列名去查找，而列表是不可哈希的类型，自然就报错了。

修正方法很简单 #

只需要去掉外层多余的方括号，直接用ds['cols']作为列选择器就行，因为它本身就是Pandas需要的列名列表格式：

df_out = df_in[ds['cols']]

完整的可运行代码 #

import pandas as pd
ds = {'cols': ['b', 'c']}
d = {'a': [2, 3], 'b': [3, 4], 'c': [4, 5]}
df_in = pd.DataFrame(data=d)
print(ds)
print(df_in)
# 修正后的列选择代码
df_out = df_in[ds['cols']]
print(df_out)

运行结果 #

{'cols': ['b', 'c']}
   a  b  c
0  2  3  4
1  3  4  5
   b  c
0  3  4
1  4  5

这样就能正确选择字典里指定的列并保存到df_out里啦！

内容的提问来源于stack exchange，提问作者John