Question

我有两个大小相同的numpy ndarray，它们具有排列的行：

import numpy as np
a = np.ndarray([[1,2,3],
                [4,5,6],
                [7,8,9],
                [10,11,12]])
b = np.ndarray([[7,8,9],
                [10,11,12],
                [1,2,3],
                [4,5,6]])

我想要一个函数，该函数返回相对于第二个数组的第一个数组中每一行的索引。例如：

compare_row_indices(a,b)

会返回

[2,3,0,1] # 0-based indexing

实现此功能的最python方式是什么？

Answer 1

也许不是最好的方法，但这似乎可行（将其分解为多个步骤以便于可视化）：

>>> cmp = a[:, None] == b
>>> cmp
array([[[False, False, False],
        [False, False, False],
        [ True,  True,  True],
        [False, False, False]],

       [[False, False, False],
        [False, False, False],
        [False, False, False],
        [ True,  True,  True]],

       [[ True,  True,  True],
        [False, False, False],
        [False, False, False],
        [False, False, False]],

       [[False, False, False],
        [ True,  True,  True],
        [False, False, False],
        [False, False, False]]])
>>> eq = np.all(cmp, axis=-1)
>>> eq
array([[False, False,  True, False],
       [False, False, False,  True],
       [ True, False, False, False],
       [False,  True, False, False]])
>>> np.argwhere(eq)
array([[0, 2],
       [1, 3],
       [2, 0],
       [3, 1]])
>>> np.argwhere(eq)[:, 1]
array([2, 3, 0, 1])

Python比较两个数组的索引与numpy中的排列行

1 个答案: