Numpy：如何檢查在ndarray中存在的元組

Question

在numpy數組中使用元組時，我發現了一個奇怪的行爲。我想得到一張布爾值表，告訴我陣列a中的哪些元組也存在於數組b中。通常，我會使用in,in1d中的任何一個。他們都沒有工作，而tuple(a[1]) == b[1,1]收益率True。

我填補我a和b這樣的：

a = numpy.array([(0,0)(1,1)(2,2)], dtype=tuple) 

b = numpy.zeros((3,3), dtype=tuple) 
for i in range(0,3): 
    for j in range(0,3): 
     b[i,j] = (i,j) 

誰能告訴我解決我的問題，請賜教爲什麼預期這不起作用？

（使用python2.7和numpy1.6.2在這裏順便說一句。）

Answer 1

爲什麼這不起作用

短的版本是，numpy的的實施array.__contains__()似乎被打破。 Python中的in運算符在幕後調用__contains__()。

含義a in b相當於b.__contains__(a)。

我在一個REPL裝載了您的陣列和嘗試以下操作：

>>> b[:,0] 
array([(0, 0), (1, 0), (2, 0)], dtype=object) 
>>> (0,0) in b[:,0] # we expect it to be true 
False 
>>> (0,0) in list(b[:,0]) # this shouldn't be different from the above but it is 
True 
>>> 

如何修復

我怎麼沒看到你的列表理解可以工作，因爲a[x]是一個元組和b[:,:]是一個二維矩陣，所以它們當然不相等。但我假設你打算使用in而不是==。如果我在這裏錯了，那麼請糾正我，你的意思是我看不到的東西。

的第一步是b從二維數組轉換成一維數組，所以我們可以通過它直線篩，並將其轉換爲一個列表，以避免numpy的殘破array.__contains()像這樣：

bb = list(b.reshape(b.size)) 

或者，更好然而，使它成爲一個set因爲元組是不可變的和一組檢查in是O（1）而不是列表的O（n）的行爲

>>> bb = set(b.reshape(b.size)) 
>>> print bb 
set([(0, 1), (1, 2), (0, 0), (2, 1), (1, 1), (2, 0), (2, 2), (1, 0), (0, 2)]) 
>>> 

接下來，我們簡單地使用列表理解導出表布爾的

>>> truth_table = [tuple(aa) in bb for aa in a] 
>>> print truth_table 
[True, True, True] 
>>> 

全碼：

def contained(a,b): 
    bb = set(b.flatten()) 
    return [tuple(aa) in bb for aa in a]