Як я можу використовувати 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, а dictскладається з key:valueелементів, ви будуєте не dictтут, а а 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 Вам не потрібно мати інше для будь-якого з них, якщо цього не хочете.
Марцін

@Marcin Чи можна для мене використовувати лише "if" для ключової частини та використовувати як "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

Якщо у вас є різні умови для оцінки ключів та значень, відповідь Марціна - це шлях.

Якщо у вас однакові умови для ключів і значень, вам краще скористатися build (ключ, значення) -парами в генератор-виразі, що подає на 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.