查找 2d 列表按值重叠的位置

Question

一个 numpy 二维数组如下所示：

[[0 1 2]
 [1 5 0]]

另一个 numpy 二维数组，如下所示：

[[0 0 0 0 0 1 1 1 1 1 1 2 2 2 2 2 2]
 [0 1 3 4 8 0 1 3 6 7 8 0 1 2 3 6 8]]

我只想得到它们“重叠”的地方：

[[0 2]
 [1 0]]

不使用 for 循环

Answer 1

您可以使用 intersect1d。 我将 n1 称为第一个数组，将 n2 称为第二个数组。

结果并不完全符合您的预期，但我相信它是正确的。

intersection = np.intersect1d(n1, n2)
print(intersection)
[0 1 2]