python - 在 pandas 数据框列(又名 pd.series)中查找数组元素位置

Question

我有一个类似于这个的 pandas 框架:

import pandas as pd
import numpy as np

data = {'Col1' : [4,5,6,7], 'Col2' : [10,20,30,40], 'Col3' : [100,50,-30,-50], 'Col4' : ['AAA', 'BBB', 'AAA', 'CCC']}

df = pd.DataFrame(data=data, index = ['R1','R2','R3','R4'])

    Col1  Col2  Col3 Col4
R1     4    10   100  AAA
R2     5    20    50  BBB
R3     6    30   -30  AAA
R4     7    40   -50  CCC

给定一个目标数组:

target_array = np.array(['AAA', 'CCC', 'EEE'])

我想在 Col4 中找到也出现在 target_array 中的单元格元素索引。

我试图找到记录在案的答案，但似乎超出了我的能力范围...有人有任何建议吗？

附言顺便说一句，对于这种特殊情况，我可以输入一个目标数组，其元素是数据帧索引名称 array(['R1', 'R3', 'R5'])。那样会更容易吗？

编辑 1:

非常感谢您的所有精彩回复。遗憾的是我只能选择一个，但每个人似乎都认为@Divakar 是最好的。您仍然应该查看 piRSquared 和 MaxU 速度比较以了解所有可用的可能性

Answer 1

您可以使用 NumPy's in1d -

df.index[np.in1d(df['Col4'],target_array)]

解释

1) 创建一个对应于每一行的 1D 掩码，告诉我们 col4's 元素与 target_array 中的任何元素是否匹配:

mask = np.in1d(df['Col4'],target_array)

2) 使用掩码从数据框中选择有效索引作为最终输出:

out = df.index[np.in1d(df['Col4'],target_array)]