python - 从 numpy 数组中获取值列表的索引

Question

像这样:

>> arr = np.array([[0, 50], [100, 150], [200, 250]]) 
>>> values = [100, 200, 300] 

>>> arr in values

预期:

array([[False, False],
       [ True, False],
       [ True, False]])

结果:

ValueError: The truth value of an array with more than one element is ambiguous. Use a.any() or a.all()

我编写了以下代码并且它可以工作，但是此代码无法接受更改列表的长度

(arr==values[0]) | (arr==values[1]) | (arr==values[2])

Answer 1

使用 np.isin:

import numpy as np

arr = np.array([[0, 50], [100, 150], [200, 250]])
values = [100, 200, 300]

np.isin(arr, values)

结果:

array([[False, False],
       [ True, False],
       [ True, False]])