查找与Python中数据框元素列表对应的索引列表

Question

我有一个包含数百万行的数据帧。 我需要获取原始数据帧中存在的数十万个元素的相应索引。

我目前使用此代码：

df[df['processed_col'] == element.index[0]

找到'element'在整个数据框中的位置。

有没有办法把一个像element1, element2,..., elementN这样的列表作为输入，这将返回一个相应索引的列表：

df[df['processed_col'] == [element1, element2, ..., elementN].index[0]

Answer 1

IIUC你需要isin如果发现指数的第一个值：

df[df['processed_col'].isin([element1, element2,..., elementN])].index[0]

或者如果想要索引的所有值，只删除[0] ：

df[df['processed_col'].isin([element1, element2,..., elementN])].index

如果需要转换为列表使用tolist ：

df[df['processed_col'].isin([element1, element2,..., elementN])].index.tolist()