python - 运算符中的 Numpy/pandas 元素

Question

我有一个数组(或者更确切地说是 pandas 框架)，其中有一列 A，该列中的值是整数(假设它们属于范围 1..10)。

现在我必须选择此数组中 A 值为 {3, 6, 9} 的行(在本例中，可以只是或 == 操作，但在现实生活中这个集合要长很多。

库(pandas 或 numpy)中是否有任何函数可以让我快速执行以下操作:

arr = pandas.DataFrame(...)
values = [3, 6, 9] 
valid_indexes = magic_function(arr.A, values)

或者在 numpy 中:

arr = np.ndarray(...)
values = [3, 6, 9] 
valid_indexes = magic_function(arr[13, :], values)

换句话说，我正在寻找元素方面的 in 运算符。

Answer 1

文档是 here

arr.loc[arr.A.isin([3,6,9])]