参考字典值

Question

我有一个数据框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