发帖
雷达卡
我有一个数据帧，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
解决办法：是的你可以。只需将列表本身传递到字典中，而不是使用列表名称的字符串。您还需要修改字典的调用（请参阅下面的说明）。试试这个：
dic={'key1':[A,B], 'key2': [C,D]}
col = df[dic['key1'][0][0]]
print(col)
说明：
您的目标是获取dataframe列df['xx']。要从'xx'字典中获取，您需要在字典中获取所需键的值。你的通话中有一个拼写错误，因为你的密钥是一个字符串，所以你需要写dic['key1']。这将返回[['xx'],['yy']]返回2个子列表的列表。从这个列表中你需要获得第一个具有索引的子列表[0]。那么，dic['key1'][0]会给你子列表['xx']。但是因为这也是一个列表而你实际上想要它的第一个元素，所以你必须再次使用索引[0]。因此，调用 dic['key1'][0][0]将为您提供所需的字符串'xx'。
备注：如果A真的只包含一个包含一个元素的列表，我建议不要使用列表而只是写A = 'xx'。然后，您可以保存第二次[0]调用字典，并且dataframe列的调用将如下所示：df[dic['key1'][0]]
京公网安备 11010802022788号
