Python-在numpy.array中查找唯一的行

Question

需要在numpy.array中找到唯一的行。

例如：

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

我知道我可以创建一个set并在数组上循环，但是我正在寻找一种有效的纯numpy解决方案。我相信有一种方法可以将数据类型设置为void，然后我可以使用numpy.unique，但我不知道如何使它工作。

Answer 1

从NumPy 1.13开始，你可以简单地选择轴来选择任何N维数组中的唯一值。要获得唯一的行，可以执行以下操作：

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