Как использовать if / else в понимании словаря?

в python2. 7 + существует любой способ сделать что-то вроде:

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

Я знаю, что вы можете сделать что-нибудь с 'если'

{ something_if_true for key, value in dict_.items() if condition}
2 62
python nsdictionary dictionary-comprehension

2 ответа:

вы уже получили его:A if test else B является допустимым выражением python. Единственная проблема с вашим пониманием dict, как показано, заключается в том, что место для выражения в понимании dict должно иметь два выражения, разделенных двоеточием:

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

финал if пункт действует как фильтр, который отличается от условного выражения.

123

@Марцина охватывает все это, но на всякий случай кто-то хочет увидеть реальный пример, я добавляю два ниже:

допустим, у вас есть следующий словарь наборов

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

теперь предположим, у вас есть два словаря, как это

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
dout2 = {d2.keys()[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'}}
3