需要在numpy.array中找到唯一的行。
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,但我不知道如何使它工作。
void
numpy.unique
从NumPy 1.13开始,你可以简单地选择轴来选择任何N维数组中的唯一值。要获得唯一的行,可以执行以下操作:
unique_rows = np.unique(original_array, axis=0)