如何檢查矩陣中是否有任何重複行？

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，然後您可以顯示您的錯誤信息