[譯] 找到ndarray中的重複行

原始數據爲:

>>> a # I have
array([[1, 1, 1, 0, 0, 0],
       [0, 1, 1, 1, 0, 0],
       [0, 1, 1, 1, 0, 0],
       [1, 1, 1, 0, 0, 0],
       [1, 1, 1, 1, 1, 0]])

需要變成的數據爲:

>>> new_a # I want to get to
array([[1, 1, 1, 0, 0, 0],
       [0, 1, 1, 1, 0, 0],
       [1, 1, 1, 1, 1, 0]])

NumPy 1.13 可以直接選取維度, 然後使用np.unique進行去重.

unique_rows = np.unique(original_array, axis=0)

axis = 0 按照行看是否重複
axis = 1 按照列看是否重複

原文鏈接: Find unique rows in numpy.array