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


138

Существует ли способ в Python 2.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}

4
как сказал @Marcin, a dictсостоит из key:valueэлементов, вы строите не dictздесь, а a set(см. набор литералов ).
mdeous

Ответы:


247

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

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

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


28
Стоит отметить, что вам не нужно иметь условие if-else как для ключа, так и для значения. Например, {(a if condition else b): value for key, value in dict.items()}будет работать.
Джереми Вейрих

5
@JeremyWeirich Вам не нужно иметь if-else для любого из них, если вы не хотите.
Марчин

@ Марчин Возможно ли мне использовать только «если» для ключевой части и использовать «если» и «еще» для значения?
nithin11

14

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

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

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

для вашего второго примера использования d1, d2я получаюAttributeError: 'dict_values' object has no attribute 'index'
alancalvitti

@alancalvitti: спасибо за указание на это! Решение было для Python 2 и не работает для Python 3; Я также добавил решение Python 3.
Клеб

3

Если у вас есть разные условия для оценки ключей и значений, ответ @ Marcin - это то, что вам нужно.

Если у вас одинаковые условия для ключей и значений, вам лучше использовать сборки (ключ, значение) в выражениях генератора, передавая в dict():

dict((modify_k(k), modify_v(v)) if condition else (k, v) for k, v in dct.items())

Его легче читать, и условие оценивается только один раз для каждого ключа, значения.

Пример с заимствованием @ словаря множеств Клеба:

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

Предположим, что вы хотите использовать суффикс только keysс aего valueи вы хотите valueзаменить длину набора в таком случае. В противном случае пара ключ-значение должна остаться без изменений.

dict((f"{k}_a", len(v)) if "a" in v else (k, v) for k, v in d.items())
# {'key1_a': 3, 'key2': {'bar', 'foo'}, 'key3': {'sad', 'so'}}

0

Другой пример использования if / else в словарном понимании

Я работаю над настольным приложением для ввода данных для моей собственной офисной работы, и такое приложение для ввода данных обычно получает все записи из виджета ввода и выгружает их в словарь для дальнейшей обработки, такой как проверка или редактирование, которое мы должны вернуть выбранные данные из файла обратно в виджеты ввода и т. д.

Первый раунд с использованием традиционного кодирования (8 строк):

entries = {'name': 'Material Name', 'maxt': 'Max Working Temperature', 'ther': {100: 1.1, 200: 1.2}}

a_dic, b_dic = {}, {}

for field, value in entries.items():
    if field == 'ther':
        for k,v in value.items():
            b_dic[k] = v
        a_dic[field] = b_dic
    else:
        a_dic[field] = value
    
print(a_dic)
 {'name': 'Material Name', 'maxt': 'Max Working Temperature', 'ther': {100: 1.1, 200: 1.2}}”

Второй раунд Я пытался использовать словарное понимание, но цикл все еще там (6 строк):

entries = {'name': 'Material Name', 'maxt': 'Max Working Temperature', 'ther': {100: 1.1, 200: 1.2}}

for field, value in entries.items():
    if field == 'ther':
        b_dic = {k:v for k,v in value.items()}
        a_dic[field] = b_dic
    else:
        a_dic[field] = value
    
print(a_dic)
 {'name': 'Material Name', 'maxt': 'Max Working Temperature', 'ther': {100: 1.1, 200: 1.2}}”

И, наконец, с помощью однострочного выражения для словаря (1 строка):

entries = {'name': 'Material Name', 'maxt': 'Max Working Temperature', 'ther': {100: 1.1, 200: 1.2}}

a_dic = {field:{k:v for k,v in value.items()} if field == 'ther' 
        else value for field, value in entries.items()}
    
print(a_dic)
 {'name': 'Material Name', 'maxt': 'Max Working Temperature', 'ther': {100: 1.1, 200: 1.2}}”

Я использую Python 3.8.3

Используя наш сайт, вы подтверждаете, что прочитали и поняли нашу Политику в отношении файлов cookie и Политику конфиденциальности.
Licensed under cc by-sa 3.0 with attribution required.