使用字典列表值选择熊猫数据框的列

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：无法散列的类型：“列表”

Answer 1

删除嵌套列表- [] ：

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

Answer 2

根据ref ，只需要放一套括号即可。

df_out = df_in[ds['cols']]