python - 从字典中引用 Pandas 数据框列标题

Question

我有一个数据框 df，其列标题为 xx、yy、zz、aaa

然后我就有了列表

A=['xx']
B=['yy']
C=['zz']

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

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

df[dic[key1[0]]]

仅给出 xx 列的输出？

   xx
0  44
1  44
2  44
3  33

Answer 1

是的，你可以。只需将列表本身传递到字典中，而不是传递带有列表名称的字符串即可。您还需要修改字典的调用(请参阅下面的说明)。试试这个:

dic={'key1':[A,B], 'key2': [C,D]}
col = df[dic['key1'][0][0]]
print(col)

说明:

您的目标是获取数据帧列df['xx']。要从字典中获取 'xx'，您需要获取字典中所需键的值。您的调用中有一个拼写错误，因为您的 key 是一个字符串，因此您需要编写 dic['key1']。这将返回列表 [['xx'],['yy']]，它返回 2 个子列表。从这个列表中，您需要获取第一个子列表，其索引为[0]。因此，dic['key1'][0] 将为您提供子列表['xx']。但由于这也是一个列表，并且您实际上想要它的第一个元素，因此您必须再次使用索引 [0]。因此，调用 dic['key1'][0][0] 将为您提供所需的字符串 'xx'。

备注:如果A确实只包含一个只有一个元素的列表，我建议不要使用列表，而只写A = 'xx'。然后，您可以在字典调用中保存第二个 [0] ，数据帧列的调用将类似于:df[dic['key1'][0]]