如何检查矩阵中是否有任何重复项？

Question

如果矩阵的第一列中有任何重复项（例如下面的列），我需要显示错误消息。

np.array([[s1,2,4,7],
          [s2,7,10,12],
          [s1,2,-3,4]])

Answer 1

您可以比较第一列np.unique的输出形状在原始列上是否相同：

In [301]:
np.unique(a[:,0]).shape == a[:,0].shape

Out[301]:
False

此处np.unique返回唯一值：

In [302]:
np.unique(a[:,0])

Out[302]:
array(['s1', 's2'], 
      dtype='<U2')

因此，如果False您可以显示错误消息