檢查如果數組的元素存在於在python

Question

另一個陣列我有兩個陣列

a = array([1,2,3])  
b = array([2,7]) 

現在我想檢查的元件是在B 和返回的答案應該是（假，真假）。有沒有一些簡單的方法來做到這一點，而不使用函數？

Answer 1

使用標準的Python列表：

>>> a = [1, 2, 3] 
>>> b = [2, 7] 
>>> tuple(x in b for x in a) 
(False, True, False) 

假設你array函數返回還支持迭代和in操作的對象，它應該工作一樣。

Answer 2

好了，這是我如何與清單做：

>>> a = [1, 2, 3] 
>>> b = [2, 7] 
>>> result = [] 
>>> 
>>> for x in a: 
... result.append(x in b) 
... 
>>> print result 
[False, True, False] 

Answer 3

只使用numpy的：從(N,)

>>> (a[:,None] == b).any(axis=-1) 

（因此，我們把a到(N,1)數組，然後使用numpy的廣播測試是否相等，最後得到一個(N, M)數組（假設b的形狀爲(M,) ...），然後我們檢查每行上是否有Trueany(axis=-1)。

Answer 4

如何：

>>> numpy.setmember1d(a, b) 
array([False, True, False], dtype=bool) 

更新，謝謝seberg。隨着numpy的較新的verions是這樣的：

>>> numpy.in1d(a, b) 
array([False, True, False], dtype=bool)