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])