在一个列表中存储多个整数值并返回最佳值对

Question

我有以下三个整数值：

id        # identifies the pair
entropy   # gives entropy information
len       # basicly the length of a string

现在我想存储其中的许多值，并选择总体上具有最高熵且长度值超过n的前10名

from collections import defaultdict

d = defaultdict(list)

for id, entropy, len in generateValues:
    d[id].append(entropy)
    d[id].append(len)

# now get the top 10 values

这可以轻松完成吗？

Answer 1

在构建完这样的字典后，您可以获得前10个值。虽然如果你在构建字典时找到它们会有更有效的解决方案，如果可能的话。

import heapq
heapq.nlargest(10, (k for k in d if d[k][1] > n), key=lambda k: d[k][0])

Answer 2

要解决您的问题，sorted支持key参数：

filtered = ((k,v) for k,v in d.iteritems() if v[1] > n) # or filter(d.iteritems(), lambda t: t[1][1] > n)
topTen = sorted(filtered, key=lambda t: t[0], reversed=true)[:10]

这是imho，比使用heapq的解决方案更具可读性（和效率相当）。