如何獲得等於另一個數組的任何值的numpy數組的指示？

Question

我有一個numpy的陣列A，都說長度100，和另一個數組B，長度爲4

的我想也是長度100的面具，這個面具將有真值，當元素A等於B的任何元素，否則等於FALSE。我怎樣才能以有效的方式做到這一點？

Answer 1

嘗試：

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

我認爲這是你想要的