Jak mogę użyć if / else w zrozumieniu słownikowym?

148

Czy w Pythonie 2.7+ istnieje sposób na wykonanie czegoś takiego jak poniżej?

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

Wiem, że możesz zrobić wszystko, używając tylko „jeśli”:

{ something_if_true for key, value in dict_.items() if condition}
diegueus9
źródło
4
jak powiedział @Marcin, a dictskłada się z key:valueelementów, nie budujesz dicttutaj, ale a set(patrz zestaw literałów ).
mdeous

Odpowiedzi:

267

Już to masz: A if test else Bto prawidłowe wyrażenie w Pythonie. Jedynym problemem związanym ze zrozumieniem dyktowania, jak pokazano, jest to, że miejsce na wyrażenie w zrozumieniu dyktowania 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() }

Ostatnia ifklauzula działa jak filtr, który różni się od wyrażenia warunkowego.

Marcin
źródło
31
Warto wspomnieć, że nie musisz mieć warunku if-else zarówno dla klucza, jak i wartości. Na przykład {(a if condition else b): value for key, value in dict.items()}zadziała.
Jeremy Weirich
5
@JeremyWeirich Nie musisz mieć if-else dla żadnego z nich, jeśli nie chcesz.
Marcin
@Marcin Czy mogę użyć tylko „if” dla części kluczowej i użyć zarówno „if”, jak i „else” dla części wartości?
nithin 11
15

@ Marcin odpowiada wszystkim, ale na wypadek gdyby ktoś chciał zobaczyć rzeczywisty przykład, poniżej dodaję dwa:

Powiedzmy, że masz następujący słownik zestawów

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

i chcesz utworzyć nowy słownik, którego klucze wskazują, czy ciąg 'a'jest zawarty w wartościach, czy nie, możesz 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()}

która 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 takie słowniki

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 zastąpić klucze w d1kluczami lub d2jeśli odpowiednie wartości są identyczne, możesz to zrobić

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

co daje

{'bad_key2': {'bar', 'foo'},
 'bad_key3': {'sad', 'so'},
 'good_key2': {'a', 'b', 'c'}}
Cleb
źródło
dla twojego drugiego przykładu d1, d2otrzymujęAttributeError: 'dict_values' object has no attribute 'index'
alancalvitti
@alancalvitti: dziękujemy za zwrócenie uwagi! Rozwiązanie było dla Pythona 2 i nie działa dla Pythona 3; Dodałem też rozwiązanie Python 3.
Cleb
3

Jeśli masz różne warunki do oceny kluczy i wartości, odpowiedz @ Marcin jest do zrobienia.

Jeśli masz ten sam warunek dla kluczy i wartości, lepiej będzie, jeśli zbudujesz (klucz, wartość) -krotki w wyrażeniu generatora przekazującym dict():

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

Jest łatwiejszy do odczytania, a warunek jest oceniany tylko raz na klucz, wartość.

Przykład z wypożyczaniem słownika zbiorów @ Cleba:

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

Załóżmy, że chcesz mieć przyrostek tylko keysz aw jego valuei chcesz valuew takim przypadku zastąpić długość zestawu. W przeciwnym razie para klucz-wartość powinna pozostać niezmieniona.

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'}}
Darkonaut
źródło
0

Kolejny przykład użycia if / else w zrozumieniu słownikowym

Pracuję nad aplikacją komputerową do wprowadzania danych do własnej pracy biurowej i często taka aplikacja do wprowadzania danych pobiera wszystkie wpisy z widżetu wejściowego i wrzuca do słownika w celu dalszego przetwarzania, takiego jak walidacja lub edycja, którą musimy zwrócić wybrane dane z pliku z powrotem do widżetów wejściowych itp.

Pierwsza runda z tradycyjnym kodowaniem (8 linii):

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

Druga runda próbowałem użyć rozumienia słownikowego, ale pętla wciąż tam jest (6 linii):

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

Na koniec za pomocą jednowierszowego wyrażenia ze słownika (1 wiersz):

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

Używam Pythona 3.8.3

KokoEfraim
źródło