在 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可以减少行或列。