参考字典值[重复]

Question

这个问题在这里已有答案：

• Selecting multiple columns in a pandas dataframe 16个答案

我有一个数据帧，df列标题为A，B，C，D

我可以从这本词典中引用这些标题吗？

dic={'key1':['A','B'],
key2: [C,D]}

df[key1[0]]

给出A列的输出？

   A
0  44
1  44
2  44
3  33

Answer 1

是的你可以，但你需要

df[dic['key1']]
     A    B
0  NaN  NaN

df[[dic['key1'][0]]]
     A
0  NaN