python - 在二维数组列中查找重复值

Question

我有一个具有以下形状的数组

(N, 2)下面是我手头的二维数组的一个例子:

[[0,2]
[0,3]
[1,2]
[1,3]
[1,4]]

我想获取第二个索引中所有重复的值。在上面的示例中，我希望返回值 2 和 3。

是否有针对此类任务的特定 np 函数？

它似乎与 np.unique 相反，但我还没有找到解决这个问题的有效函数。

Answer 1

您可以对第二列进行索引并使用 np.bincount 查找计数高于 1 的索引:

a = np.array([[0,2],
            [0,3],
            [1,2],
            [1,3],
            [1,4]])

np.flatnonzero(np.bincount(a[:,1])>1)
# array([2, 3], dtype=int64)

或者对于大整数，np.unique 可能是更好的选择:

u, c = np.unique(a[:,1], return_counts=True)
u[c>1]
# array([2, 3])