根據第二個元組的第二個索引找到k個最小的值

Question

我需要根據它們在python中的第二個索引找到k個最小的元組值，例如，我有一個元組列表，其中包含距固定點某些點的距離，例如： [('p1',12.5),('p2',3),('p4',1),('p5',16),('p6',15),...](from fixed point p0) ，我想根據到距離。

我的意思是這樣的代碼，在這里解釋如下： 查找numpy數組的k個最小值

謝謝您的解決方案

Answer 1

您可以使用heapq.nsmallest ，並將元組中的第二個元素指定為比較鍵：

from heapq import nsmallest 

lst = [('p1',12.5),('p2',3),('p4',1),('p5',16),('p6',15)]
nsmallest(3, lst, key=lambda x: x[1])
# [('p4', 1), ('p2', 3), ('p1', 12.5)]