Trouver des index par numpy où

Question

J'ai une liste de tableaux numpy :

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

Pour trouver des index de "A" J'utilise np.where(x == "A") et cela fonctionne bien. Comment puis-je trouver les index de "A" y "B" ?

J'ai utilisé :

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

Mais ça ne m'a pas aidé.

Answer 1

Utilisez | o np_logical_or pour un OU par éléments :

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