热线电话:13121318867

登录
2019-02-25 阅读量: 781
从字典中引用Pandas dataframe列标题

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

0.0000
2
关注作者
收藏
评论(0)

发表评论

暂无数据
推荐帖子