通过 numpy where 查找索引

Question

我有 numpy 数组列表：

x = ["A", "A", "B", "A", "C", "D", "B"]

为了查找 "A" 的索引，我使用 np.where(x == "A") 并且它工作正常。如何找到 "A" 和 "B" 的索引？

我用过：

np.where(x == ["A", "B"]) 
np.where(x in ["A", "B"])
np.where(x == any(["A", "B"]))

但这对我没有帮助。

Answer 1

使用 | 或 np_logical_or 进行元素或：

x = np.array(x)
np.where((x=='A') | (x=='B'))
Out: (array([0, 1, 2, 3, 6], dtype=int64),)

np.where(np.logical_or(x=='A', x=='B'))
Out: (array([0, 1, 2, 3, 6], dtype=int64),)

通过 numpy where 查找索引

Find indexes by numpy where

python

numpy

where