Python的Numpy intersect1d與一維數組與二維數組

Question

我試圖用二維數組上的視圖intersect1d函數來找到它的第一列與另一個1D數組的交集，維護其最終結果中的第二列。不過，我試圖構建視圖卡住了。

示例輸入：

a1 = np.array([[1,2],[3,4],[4,9]], dtype=np.uint32) 
a2 = np.array([8,3,8,1,0,9,3,2], dtype=np.uint32) 

期望的結果是：

[[1,2],[3,4]] 

Answer 1

理想情況下，希望有的行指數或交叉那些的掩模。現在np.intersect1d實際上也不會給你。要解決這個問題，你可以使用np.in1d來獲取相交行的蒙版。因此，索引與這將是所需輸出，像這樣 -

a1[np.in1d(a1[:,0],a2)] 

樣品運行 -

In [15]: a1 
Out[15]: 
array([[1, 2], 
     [3, 4], 
     [4, 9]], dtype=uint32) 

In [16]: a2 
Out[16]: array([8, 3, 8, 1, 0, 9, 3, 2], dtype=uint32) 

In [17]: np.in1d(a1[:,0],a2) # Intersecting rows mask for a1 
Out[17]: array([ True, True, False], dtype=bool) 

In [18]: a1[np.in1d(a1[:,0],a2)] 
Out[18]: 
array([[1, 2], 
     [3, 4]], dtype=uint32)