在兩個列表中找到k個最小的對

Question

我的目標是：

1. 將兩個列表中的元素配對； 和
2. 找到k最小的對。

目前，我可以將兩個列表中的每個元素與以下代碼配對：

import itertools  

a = [1, 3, 5, 7]
b = [2, 4, 6, 8]
c = list(itertools.product(a, b))

print c

我的輸出是：

[(1, 2), (1, 4), (1, 6), (1, 8), (3, 2), (3, 4), (3, 6), (3, 8), (5, 2), (5, 4), (5, 6), (5, 8), (7, 2), (7, 4), (7, 6), (7, 8)]

假設我設置k = 3 ，它應該返回給我三個最小的對： (1, 2) ， (1, 4)和(3, 2) 。 我怎么做？

Answer 1

您可以使用帶有sum heapq.nsmallest作為其關鍵功能：

>>> import heapq
>>> heapq.nsmallest(3,c,key=sum)
[(1, 2), (1, 4), (3, 2)]

或者如@jonrsharpe在評論中所說，您可以使用sorted ：

sorted(c, key=sum)[:k]