在数据框中搜索值向量

2024-10-03 13:20:09 发布

您现在位置:Python中文网/ 问答频道 /正文

我想在数据帧的一列中搜索几个值

列中的值是唯一的。你知道吗

不是所有的值都必须存在,在这种情况下,它应该返回一些None-1或其他什么。你知道吗

举个例子:

ddff = pd.DataFrame({'A':['10A','20B','30A','40T','G50'],
                     'B':[15,25,3,6,10]},
                     index = ['aa','bb','cc','dd','ee'])
look_for = ['10A','20ZZZ','G50','30A']

我总是可以遍历look_for并搜索每个值。你知道吗

如何应用搜索,使结果成为每个值(如果存在)的索引(因为我知道列'A'具有uniqu值),从而高效地获得以下结果:

['aa',None,'ee','cc']

Tags: 数据nonedataframeforindex情况ee例子
1条回答
网友
1楼 · 发布于 2024-10-03 13:20:09

我想可以使用列表理解来完成,这样就可以保持look_for变量的顺序:

[ddff.A.loc[ddff.A == i].index[0] if i in ddff['A'].values else None for i in look_for]
Out[7]: ['aa', None, 'ee', 'cc']

相关问题 更多 >