排序列表后如何保存原始索引?

时间:2020-04-01 11:37:55

标签: python list unordered

假设我有以下数组:

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

然后我对其进行排序:

b = sorted(A) = [1,2,3,4,4]

我怎么能找到一个列表,列出每个数字的位置,例如:

position(b,a) = [3,1,2,0,4]

以澄清此列表包含的位置不是值)

(ps还考虑到前4个位于位置0)

2 个答案:

答案 0 :(得分:7)

b = sorted(enumerate(a), key=lambda i: i[1])

这将产生一个元组列表,其第一项是原始索引,第二项是值:

[(3, 1), (1, 2), (2, 3), (0, 4), (4, 4)]

答案 1 :(得分:2)

def position(a):
    return sorted(range(len(a)), key=lambda k: a[k])