Pythonジェネレータ式内の値をフィルタリングする

Question

辞書dctがあります。対応するキーが指定されたリストlstに存在すれば、それぞれの値を合計したいと思っています。

私がこれまで使っていたコードは次のとおりです。私は、リストからキーが辞書の中に発見されていない場合にはKeyErrorを処理したいと思います上記のジェネレータ式で

sum(dct[k] for k in lst) 

。私はtry - exceptのアプローチを実装する方法や、このジェネレータの表現の中でif - elseのアプローチを（構文的に）実装する方法を見つけることができないようです。

リスト内のキーが辞書内に見つからない場合は、他の値を取得する必要があります。 合計の最終結果は、キーが見つからない場合、の影響を受けてはなりません。いずれのキーも存在しない場合、ゼロは和の結果でなければならない。

Answer 1

まあ、いくつかのオプションがあり、1が使用するのが好ましいdict.get()：

# 1 
sum(dct.get(k, 0) for k in lst) 
# 2 
sum(dct[k] for k in lst if k in dct) 

オプションの1つは、それを反復する前にlstをフィルタリングすることです。

sum(dct[k] for k in filter(lambda i: i in dct, lst)) 

そして、あなたはsumの代替としてフィルタリングされたリストにreduce functionを使用することができます。

reduce(lambda a, k: a + dct[k], filter(lambda i: i in dct, lst)) 

今度はtimeitで最速のアプローチを見つけてみましょう：

from timeit import timeit 
import random 

lst = range(0, 10000) 
dct = {x:x for x in lst if random.choice([True, False])} 

via_sum = lambda:(sum(dct.get(k, 0) for k in lst)) 
print("Via sum and get: %s" % timeit(via_sum, number=10000)) 
# Via sum and get: 16.725695848464966 

via_sum_and_cond = lambda:(sum(dct[k] for k in lst if k in dct)) 
print("Via sum and condition: %s" % timeit(via_sum_and_cond, number=10000)) 
# Via sum and condition: 9.4715681076 

via_reduce = lambda:(reduce(lambda a, k: a + dct[k], filter(lambda i: i in dct, lst))) 
print("Via reduce: %s" % timeit(via_reduce, number=10000)) 
# Via reduce: 19.9522120953 

だから、最速のオプションは、内のif文を経由してアイテムを合計することですジェネレータ表現

sum(dct[k] for k in lst if k in dct) # Via sum and condition: 9.4715681076 

グッドリックk！

Answer 2

sum(dct[k] for k in lst if k in dct) 

Answer 3

あなたは、単に辞書のキーの値を取得しようとする.get()を使用することができ、そしてこの場合には0になりNone、またはご提供するデフォルトのパラメータを返します見つけるべきではありません。

>>> dct = {1:2, 3:4, 5:6} 
>>> lst = [1,5] 
>>> sum(dct.get(k, 0) for k in lst) 
8  

は、キーの一部（または全部）が存在しない場合には、合計はまだ正常に機能します。

>>> lst = [10, 11] 
>>> sum(dct.get(k, 0) for k in lst) 
0 

Answer 4

は、次の2つのオプションがあります。キーは

sum(dct[k] for k in lst if k in dct) 

または使用して存在するかどうかのチェック

をdct.get(k, 0)戻っdct[k]がk場合dctかでキーをあるget

sum(dct.get(k, 0) for k in lst) 

0もしそうでなければ。見つからない場合は

Answer 5

あなたはデフォルト値を提供するために、辞書のget -methodを使用することができます。

sum(dct.get(k, 0) for k in lst)