我有一个数组(或者更确切地说是 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 运算符。
最佳答案
文档是 here
arr.loc[arr.A.isin([3,6,9])]
关于python - 运算符中的 Numpy/pandas 元素，我们在Stack Overflow上找到一个类似的问题： https://stackoverflow.com/questions/25297038/