Python:如何删除Nx3 numpy数组中的重复项

时间:2018-07-04 00:27:10

标签: python arrays numpy duplicates

我有Nx3 numpy数组,可以这样说:

a=[[1,1,1],[1,2,3],...,[2,1,3],[2,2,2]]

就我而言,我不在乎元素在“ sub 3D数组”中的位置,而是将它们视为重复项:

[1,2,3] == [2,1,3] == [3,1,2] = ...

我想删除这些重复项,然后得到:

a_new = [[1,1,1],[1,2,3],...,[2,2,2]]

问题是我不知道该怎么做。

欢迎任何帮助,并在此先感谢:)

1 个答案:

答案 0 :(得分:2)

使用sortunique

import numpy as np
a=np.array([[1,1,1],[1,2,3],[2,1,3],[2,2,2]])
np.unique(np.sort(a, axis=1), axis=0)

array([[1, 1, 1],
       [1, 2, 3],
       [2, 2, 2]])