python - numpy 唯一没有排序

Question

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

numpy.unique with order preserved (7 个回答)

关闭 6年前。

如何使用 numpy unique 而不对结果进行排序，而只是按照它们在序列中出现的顺序？像这样？

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

np.unique(a) = [4,2,1,3]

而不是

np.unique(a) = [1,2,3,4]

使用简单的解决方案应该可以编写一个简单的函数。但是由于我需要多次执行此操作，是否有任何快速简洁的方法来执行此操作？

Answer 1

您可以使用 return_index 参数来做到这一点:

>>> import numpy as np>>> a = [4,2,1,3,1,2,3,4]>>> np.unique(a)array([1, 2, 3, 4])>>> indexes = np.unique(a, return_index=True)[1]>>> [a[index] for index in sorted(indexes)][4, 2, 1, 3]