比较两个不同大小的数组 - python numpy

Question

所以我有两个数组，它们具有相同的维度但不同的长度。

Arr1 = np.array([[Ind1],[Ind2],[Ind3]])

Arr2 = np.array([[Ind7],[Ind3],[Ind3],[Ind4]])

我需要获取两个数组中具有相同位置且相等的元素的位置和值。

在示例情况下，预期答案将是：

位置 = 2

值 = Ind3

我正在使用 python 和 numpy 模块。

Answer 1

使用 NumPy 数组，您可能希望以矢量化方式工作以提高性能并利用数组切片。考虑到这一点，这是输入数组的一种a方法b-

n = min(len(a), len(b))
out_idx = np.flatnonzero(a[:n] == b[:n])
out_val = a[out_idx] # or b[out_idx] both work

这可以处理多个匹配。

样本运行 -

In [224]: a = np.array([3, 8, 9, 2, 1, 7])

In [225]: b = np.array([1, 2, 9, 7, 5, 7, 0, 4])

In [226]: n = min(len(a), len(b))
     ...: out_idx = np.flatnonzero(a[:n] == b[:n])
     ...: out_val = a[out_idx]
     ...: 

In [227]: out_idx
Out[227]: array([2, 5])

In [228]: out_val
Out[228]: array([9, 7])

对于作为索引及其值的输出的元组列表 -

In [229]: zip(out_idx, out_val)
Out[229]: [(2, 9), (5, 7)]

对于索引和相应值的漂亮字典输出 -

In [230]: {i:j for i,j in zip(out_idx, out_val)}
Out[230]: {2: 9, 5: 7}