在numpy.Array中找到唯一的行

Question

在numpy.Array中找到唯一的行我需要在numpy.array.例如：>>> a # I havearray([[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 toarray([[1, 1, 1, 0, 0, 0],        [0, 1, 1, 1, 0, 0],        [1, 1, 1, 1, 1, 0]])我知道我可以在数组上创建一个集合和循环，但是我正在寻找一个高效的纯numpy解决办法。我相信有一种方法可以将数据类型设置为无效，然后我可以使用numpy.unique但我想不出该怎么做。

Answer 1

从NumPy 1.13开始，可以简单地选择任意N-DIM数组中唯一值的轴。要获得唯一的行，可以这样做：

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

Answer 2

另一种可能的解决办法

np.vstack({tuple(row) for row in a})