Jak używać if/else w słowniku ze zrozumieniem?

Question

W python2.7 + istnieje jakikolwiek sposób, aby coś takiego:

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

wiem, można zrobić coś z zaledwie „if”

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

Answer 1

Masz już go: A if test else B jest prawidłowe wyrażenie Pythona. Jedyny problem z dict zrozumienia jak pokazane jest to, że miejsce dla wyrażenia w zrozumieniu dict musi mieć dwa wyrażenia oddzielone dwukropkiem:

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

Ostateczna if klauzula działa jak filtr, który różni się od posiadające wyrażenie warunkowe.

Answer 2

@Marcin's answer obejmuje to wszystko, ale tylko w przypadku, gdy ktoś chce zobaczyć rzeczywisty przykład, dodaję dwa poniżej:

Powiedzmy masz następujący słownika zestawów

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

i chcesz utworzyć nowy słownik, którego klawisze wskazują, czy ciąg 'a' jest zawarta w wartości lub nie, można użyć

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()} 

co daje

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

Teraz załóżmy, że masz dwa słowniki jak ten

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'}} 

i chcesz wymienić klucze w d1 przez klawiszy d2 jeśli odpowiednie wartości są identyczne, można zrobić

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

co daje

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