我有一个数据帧,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]]








暂无数据