如何用Python实现获取乱序列表排序后的新索引

时间：2023-05-10 11:36

对一个列表list而言，进行排序是很简单的。正序排序（从小到大）用

list.sort()

倒序排序（从大到小）用

list.sort(reverse=True)

但是如果不仅限于得到一个排完序的列表，还希望记录原本的下标，那么对于一个numpy.array而言，可以用

np.argsort()

比如[1,3,2,5,6]

排序后变成[1,2,3,5,6]

但是我们想知道排序后的结果对应原本的下标是什么（答案是[0,2,1,3,4]），就可以用np.argsort()

但是如果就是个简单的list，想实现这个效果，就可以用

# enumerate(x)会自动构造一个tuple(a,b)# 其中a是index，b是list里index下标对应的具体的值,后面的x是代表一个虚拟变量，即tuple(a,b)sorted_list = sorted(enumerate(list), key=lambda x:x[1])  # x[1]是因为在enumerate(a)中，a数值在第1位result = [x[0] for x in sorted_list]

这样返回的result就是原本的index

如果要实现标题里的，更进一步，即要得到一个list排序后的新下标，比如对于 [1,5,2,8,3,4]，应该得到[0,4,1,5,2,3]

那就可以用

# enumerate(x)会自动构造一个tuple(a,b)# 其中a是index，b是list里index下标对应的具体的值,后面的x是代表一个虚拟变量，即tuple(a,b)# sorted_list = [(0,1),(2,2),(4,3),(5,4),(1,5),(3,8)]sorted_list = sorted(enumerate(list), key=lambda x:x[1])  # x[1]是因为在enumerate(a)中，a数值在第1位for i in range(len(sorted_list)):    list[sorted_list[i][1]] = i

以上就是如何用Python实现获取乱序列表排序后的新索引的详细内容，更多请关注Gxl网其它相关文章！