在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})