python - 如何删除重复的数组元素并在python中保持第一次出现？

Question

这个问题在这里已经有了答案:





numpy.unique with order preserved

(7 个回答)


去年关闭。




我有以下大小为 (5,2) 的 numpy 数组:

A = [[5,6]
     [4,3]
     [2,1]
     [4,3]
     [8,9]]

我想删除任何重复的行(因此在本例中为 [4,3])并保留第一次出现并返回如下所示:

A = [[5,6]
     [4,3]
     [2,1]
     [8,9]]

     

Answer 1

函数 numpy.unique 可以帮助您从数组中提取唯一值。指定轴参数以选择要操作的轴。
尝试这个

np.unique(A, axis=0)

提取唯一的行。
numpy.unique 将返回一个排序的数组。如果要检索原始顺序，可以使用 return_index=True 参数返回唯一元素首次出现的索引，然后从原始数组中获取这些元素。
对于您的示例，这将起作用:

A[np.sort(np.unique(a, return_index=True, axis=0)[1])]