您可以使用numpy unique
。由于您需要唯一的行,因此我们需要将它们放入元组中:
import numpy as np
data = np.array([[1,8,3,3,4],
[1,8,9,9,4],
[1,8,3,3,4]])
刚刚申请np.unique
to the data
数组将导致这样的结果:
>>> uniques
array([1, 3, 4, 8, 9])
打印出列表中的唯一元素。因此,将它们放入元组中会导致:
new_array = [tuple(row) for row in data]
uniques = np.unique(new_array)
打印:
>>> uniques
array([[1, 8, 3, 3, 4],
[1, 8, 9, 9, 4]])
UPDATE
在新版本中,需要设置np.unique(data, axis=0)