使用where子句查找多个值的索引

Question

np.where(A ==2)[0]

给出A的索引，其中元素等于2.

如何推广可能的值列表？

我正在寻找类似的东西：

np.where(A in ([2,3,6,8]))[0]

Answer 1

自NumPy 1.13以来，您可以使用isin功能。

在以前的版本中有in1d。

测试：

A = np.array([1, 2, 3, 4, 5])
print(np.isin(A, [2, 3, 6, 8]))

结果：

[False  True  True False False]