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

Question

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

我目前使用这个代码：

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

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

有没有办法将像element1, element2,..., elementN 这样的列表作为输入，而不是做一个很长的循环，这将返回一个相应索引的列表：

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

Answer 1

如果找到索引的第一个值，您需要 isinIIUC：

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()