首页  >  文章  >  后端开发  >  如何用Python实现获取乱序列表排序后的新索引

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

WBOY
WBOY转载
2023-05-09 23:05:281749浏览

对一个列表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实现获取乱序列表排序后的新索引的详细内容。更多信息请关注PHP中文网其他相关文章!

声明:
本文转载于:yisu.com。如有侵权,请联系admin@php.cn删除