私はnumpyの配列リストを持っている

Question

numpyのことでインデックスを検索します。 "A"と"B"のインデックスはどのようにして見つけることができますか？

は、私が使用：

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

しかし、それは私を助けていませんでした。要素単位のORため

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),)