首頁  >  文章  >  後端開發  >  如何用Python實作取得亂序列表排序後的新索引

如何用Python實作取得亂序列表排序後的新索引

WBOY
WBOY轉載
2023-05-09 23:05:281810瀏覽

對一個清單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刪除