[译] 找到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