【发布时间】: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函数。
假设我有以下数组:
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)
【问题讨论】:
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]))
def position(a):
return sorted(range(len(a)), key=lambda k: a[k])
【讨论】: