存儲多個整數值在一個列表，並返回最佳值對

Question

我有以下三個整數值：

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

現在我想要存儲的許多設置值，選擇排名前10位總具有最高熵在n

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的解決方案。