You can use numpy unique
. Since you want the unique rows, we need to put them into tuples:
import numpy as np
data = np.array([[1,8,3,3,4],
[1,8,9,9,4],
[1,8,3,3,4]])
just applying np.unique
to the data
array will result in this:
>>> uniques
array([1, 3, 4, 8, 9])
prints out the unique elements in the list. So putting them into tuples results in:
new_array = [tuple(row) for row in data]
uniques = np.unique(new_array)
which prints:
>>> uniques
array([[1, 8, 3, 3, 4],
[1, 8, 9, 9, 4]])
UPDATE
In the new version, you need to set np.unique(data, axis=0)