web-dev-qa-db-ja.com

辞書内包表記でif / elseを使用するには?

Python2.7 +では、次のようなものを作成する方法があります。

{ something_if_true if condition else something_if_false for key, value in dict_.items() }

「if」だけで何でもできる

{ something_if_true for key, value in dict_.items() if condition}
96
diegueus9

あなたはすでにそれを持っています:A if test else Bは有効なpython式です。示されている辞書理解の唯一の問題は、辞書理解の式の場所にはコロンで区切られた2つの式が必要であるということです。

{ (some_key if condition else default_key):(something_if_true if condition 
          else something_if_false) for key, value in dict_.items() }

最後のif句はフィルターとして機能しますが、これは条件式を持つこととは異なります。

175
Marcin

@ Marcin's answer はすべてをカバーしていますが、実際の例をご覧になりたい場合に備えて、以下に2つ追加します。

次のセットの辞書があるとしましょう

d = {'key1': {'a', 'b', 'c'}, 'key2': {'foo', 'bar'}, 'key3': {'so', 'sad'}}

文字列'a'が値に含まれているかどうかを示すキーを持つ新しい辞書を作成したい場合は、次を使用できます。

dout = {"a_in_values_of_{}".format(k) if 'a' in v else "a_not_in_values_of_{}".format(k): v for k, v in d.items()}

もたらす

{'a_in_values_of_key1': {'a', 'b', 'c'},
 'a_not_in_values_of_key2': {'bar', 'foo'},
 'a_not_in_values_of_key3': {'sad', 'so'}}

ここで、次のような2つの辞書があるとします

d1 = {'bad_key1': {'a', 'b', 'c'}, 'bad_key2': {'foo', 'bar'}, 'bad_key3': {'so', 'sad'}}
d2 = {'good_key1': {'foo', 'bar', 'xyz'}, 'good_key2': {'a', 'b', 'c'}}

それぞれの値が同一である場合、d1のキーをd2のキーに置き換えたい場合は、

# here we assume that the values in d2 are unique
# Python 2
dout2 = {d2.keys()[d2.values().index(v1)] if v1 in d2.values() else k1: v1 for k1, v1 in d1.items()}

# Python 3
dout2 = {list(d2.keys())[list(d2.values()).index(v1)] if v1 in d2.values() else k1: v1 for k1, v1 in d1.items()}

与える

{'bad_key2': {'bar', 'foo'},
 'bad_key3': {'sad', 'so'},
 'good_key2': {'a', 'b', 'c'}}
7
Cleb