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

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])