返回带有一些重复数字的排序列表的索引

【问题标题】：Return index of a sorted list with some reoccurring umbers返回带有一些重复数字的排序列表的索引
【发布时间】：2021-09-19 04:26:19
【问题描述】：

我有一个列表，比如说 [2,3,3,4]，我想将列表的索引返回为 [0, 1, 1, 3]

【问题讨论】：

你看过np.argsort吗？
是的，它不会为重复的数字提供相同的索引。

标签： list numpy sorting

【解决方案1】：

您可以使用np.unique 实现此目的，如下所示：

import numpy as np

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

us, inds, inv = np.unique(a, return_index=True, return_inverse=True)

res = inds[inv]

assert (b == res).all()

【讨论】：