如何在Python中获得排序数组的索引

我有一个数字列表:

myList = [1, 2, 3, 100, 5]

现在如果我对这个列表进行排序得到[1,2,3,5,100]。我想要的是元素的下标原始列表的排序顺序，即[0,1,2,4,3] ——ala MATLAB的排序函数，返回两者值和索引。

当前回答

将numpy导入为np

对指数

S=[11,2,44,55,66,0,10,3,33]

r=np.argsort(S)

[output]=array([5, 1, 7, 6, 0, 8, 2, 3, 4])

argsort按顺序返回S的索引

价值

np.sort(S)

[output]=array([ 0,  2,  3, 10, 11, 33, 44, 55, 66])

其他回答

myList = [1, 2, 3, 100, 5]    
sorted(range(len(myList)),key=myList.__getitem__)

[0, 1, 2, 4, 3]

s = [2, 3, 1, 4, 5]
print([sorted(s, reverse=False).index(val) for val in s])

对于一个有重复元素的列表，它将返回不带联系的秩，例如。

s = [2, 2, 1, 4, 5]
print([sorted(s, reverse=False).index(val) for val in s])

[1, 1, 0, 3, 4]

其他答案都是错误的。

运行一次argsort并不是解决方案。例如，以下代码:

import numpy as np
x = [3,1,2]
np.argsort(x)

生成数组([1,2,0]，dtype=int64)，这不是我们想要的。

答案应该是运行argsort两次:

import numpy as np
x = [3,1,2]
np.argsort(np.argsort(x))

按预期给出数组([2,0,1]，dtype=int64)。

像下面这样:

>>> myList = [1, 2, 3, 100, 5]
>>> [i[0] for i in sorted(enumerate(myList), key=lambda x:x[1])]
[0, 1, 2, 4, 3]

enumerate(myList)给你一个包含(index, value)元组的列表:

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

通过将列表传递给sorted并指定一个函数来提取排序键(每个元组的第二个元素;这就是的意义。最后，使用[i[0] for i in…列表理解。

我们将创建另一个从0到n-1的索引数组然后压缩到原始数组，然后根据原始值对其排序

ar = [1,2,3,4,5]
new_ar = list(zip(ar,[i for i in range(len(ar))]))
new_ar.sort()

推荐文章