【问题标题】:How can I save the original index after sorting a list?排序列表后如何保存原始索引?
【发布时间】:2026-01-02 00:00:02
【问题描述】:

假设我有以下数组:

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)

【问题讨论】:

  • 可以使用numpy的argsort函数。

标签: python list unordered


【解决方案1】:
b = sorted(enumerate(a), key=lambda i: i[1])

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

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

【讨论】:

  • 我可能会想更进一步并解包(如果元组没问题),例如:orig_pos, b = zip(*sorted(enumerate(a), key=lambda i: i[1]))
  • 当然,如果想法是有两个单独的列表;但 IMO 将数据放在一起通常更有意义。
  • 是的......一切都取决于......只是把它扔在那里:)
【解决方案2】:
def position(a):
    return sorted(range(len(a)), key=lambda k: a[k])

【讨论】: