在 2 个不同大小的 numpy 数组中交叉行

Question

假设我有 2 个 numpy 数组；arr1 = np.array([       [1, 2, 3, 4],       [2, 3, 1, 4],       [2, 4, 1, 5],       ...)arr2 = np.array([       [2, 4, 1, 5],       [2, 1, 3, 5],       [1, 2, 3, 4],       ...)arr1我想获得和的行的交集arr2。我努力了：intersect = np.intersect1d(arr1, arr2)它返回array([1, 2, 3, 4, 5])这意味着按元素相交。我想按行完成这个。它应该返回；array(      [2, 4, 1, 5],      [1, 2, 3, 5])

Answer 1

In [3]: arr1 = np.array([ 

   ...:        [1, 2, 3, 4], 

   ...:        [2, 3, 1, 4], 

   ...:        [2, 4, 1, 5], 

   ...:        ]) 

   ...:  

   ...: arr2 = np.array([ 

   ...:        [2, 4, 1, 5], 

   ...:        [2, 1, 3, 5], 

   ...:        [1, 2, 3, 4], 

   ...:        ]) 

                                                                                        

广播平等后跟all和的适当组合any：

In [8]: (arr1[:,None,:]==arr2[None,:,:]).shape                                                       

Out[8]: (3, 3, 4)

In [9]: (arr1[:,None,:]==arr2[None,:,:]).all(axis=2)                                                 

Out[9]: 

array([[False, False,  True],

       [False, False, False],

       [ True, False, False]])

In [10]: (arr1[:,None,:]==arr2[None,:,:]).all(axis=2).any(axis=0)                                    

Out[10]: array([ True, False,  True])

In [12]: arr1[_]                                                                                     

Out[12]: 

array([[1, 2, 3, 4],

       [2, 4, 1, 5]])

带套

In [19]: set([tuple(row) for row in arr1])                                                           

Out[19]: {(1, 2, 3, 4), (2, 3, 1, 4), (2, 4, 1, 5)}

In [20]: set([tuple(row) for row in arr2])                                                           

Out[20]: {(1, 2, 3, 4), (2, 1, 3, 5), (2, 4, 1, 5)}

In [21]: _19.intersection(_20)                                                                       

Out[21]: {(1, 2, 3, 4), (2, 4, 1, 5)}

===

如果我扩展arr2到 4 行：

...: arr2 = np.array([ 

...:        [2, 4, 1, 5], 

...:        [2, 1, 3, 5], 

...:        [1, 2, 3, 4], 

...:        [1, 1, 1, 1], 

...:        ]) 

In [34]: (arr1[:,None,:]==arr2[None,:,:]).all(axis=2).any(axis=0)                                    

Out[34]: array([ True, False,  True, False])

anyon 0 产生一个 4 元素数组，它必须用于索引arr2（不像arr1我原来那样）：

In [35]: arr2[_]                                                                                     

Out[35]: 

array([[2, 4, 1, 5],

       [1, 2, 3, 4]])

或者any沿着另一个轴：

In [36]: (arr1[:,None,:]==arr2[None,:,:]).all(axis=2).any(axis=1)                                    

Out[36]: array([ True, False,  True])

In [37]: arr1[_]                                                                                     

Out[37]: 

array([[1, 2, 3, 4],

       [2, 4, 1, 5]])

产生all（在本例中）一个 (3,4) 数组：

In [38]: (arr1[:,None,:]==arr2[None,:,:]).all(axis=2)                                                

Out[38]: 

array([[False, False,  True, False],

       [False, False, False, False],

       [ True, False, False, False]])

any可以减少行或列。