一日一学:返回排序好的列表的索引顺序
總第 118 篇文章,本文大約?700 字,閱讀大約需要? 2?分鐘
今天介紹的是對列表排序后,返回排序好的索引順序。
問題描述:給定一個列表 [2, 3, 1, 4, 5] ,怎么返回排序后的索引順序,即 [2,0,1,3,4] ?
解決方法:
方案1: 利用 sorted 函數(shù),例子如下所示:
>>> s = [2, 3, 1, 4, 5] >>> sorted(range(len(s)), key=lambda k: s[k]) [2, 0, 1, 3, 4] >>>方案2: 采用 itemgetter 函數(shù):
L = [2,3,1,4,5] from operator import itemgetter indices, L_sorted = zip(*sorted(enumerate(L), key=itemgetter(1))) list(L_sorted) >>> [1, 2, 3, 4, 5] list(indices) >>> [2, 0, 1, 3, 4]方案3: 利用 numpy 的 argsort 函數(shù):
>>> import numpy >>> vals = numpy.array([2,3,1,4,5]) >>> vals array([2, 3, 1, 4, 5]) >>> sort_index = numpy.argsort(vals) >>> sort_index array([2, 0, 1, 3, 4])參考文章:
https://stackoverflow.com/questions/7851077/how-to-return-index-of-a-sorted-list?answertab=active#tab-top
也可以點(diǎn)擊原文查看。
ps. 今天也祝大家情人節(jié)快樂
單身的2020年脫單成功!
? ? 精選文章
python版代碼整潔之道
python技巧(1)--如何轉(zhuǎn)換itertools.chain對象為數(shù)組
python技巧(2)--碾平列表和列表去重
幾個有趣的python技巧
10個高效的pandas技巧
歡迎關(guān)注我的微信公眾號--算法猿的成長,或者掃描下方的二維碼,大家一起交流,學(xué)習(xí)和進(jìn)步!
如果覺得不錯,在看、轉(zhuǎn)發(fā)就是對小編的一個支持!
總結(jié)
以上是生活随笔為你收集整理的一日一学:返回排序好的列表的索引顺序的全部內(nèi)容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: 操作系统基础:进程知识笔记(一)
- 下一篇: 最常问的MySQL面试题集合