Python numpy.ndarray 比较

Question

如何比较 numpy.ndarray 中的数组元素？

我的ndarray是这样的

array([[ 781,   94],
       [ 781,  656],
       [1367,   94],
       [1367,  656]])

所需输出：

array([781, 94, 656, 1367])

Answer 1

您似乎希望数组中的值具有唯一性

In [16]: arr = np.array([[ 781,   94],
       [ 781,  656],
       [1367,   94],
       [1367,  656]])

In [17]: np.unique(arr)
Out[17]: array([  94,  656,  781, 1367])