使用字典列表值选择pandas dataframe的列

Question

我在字典中有列名，我想从数据帧中选择这些列。在

在下面的示例中，如何选择字典值“b”、“c”并将其保存到df1中？在

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)

TypeError: unhashable type: 'list'

Answer 1

根据ref，只需要去掉一组括号。在

df_out = df_in[ds['cols']]

Answer 2

删除嵌套列表-[]：

df_out = df_in[ds['cols']]
print(df_out)
   b  c
0  3  4
1  4  5