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

Question

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

我目前使用此代码：

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

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

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

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

Answer 1

如果找到索引的第一个值，则需要isin的IIUC：

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