Wie erhalten Sie Indizes von numpy Array, die alle Werte eines anderen Arrays entsprechen?

Question

Ich habe eine numpy Array A, sagt die Länge 100 und eine andere Anordnung B der Länge 4.

ich eine Maske von auch Länge 100, und dieser Maske Werte TRUE haben wird, wenn das Element erhalten möge von A entspricht ANY der Elemente von B, und andernfalls FALSE. Wie kann ich das effizient machen?

Answer 1

Versuchen:

a = np.random.rand(100) 
b = np.random.rand(4) 
print([(element in b) for element in a]) 
print('\n') 
print([(element in a) for element in b]) 

Ich denke, das ist, was Sie