Jak usunąć elementy ze słownika podczas iteracji?

Czy legalne jest usuwanie elementów ze słownika w Pythonie podczas iteracji?

Na przykład:

for k, v in mydict.iteritems():
   if k == val:
     del mydict[k]

Chodzi o to, aby usunąć ze słownika elementy niespełniające określonych warunków, zamiast tworzyć nowy słownik, który jest podzbiorem iterowanego.

Czy to dobre rozwiązanie? Czy istnieją bardziej eleganckie / wydajne sposoby?

EDYTOWAĆ:

Ta odpowiedź nie będzie działać dla Python3 i da RuntimeError.

RuntimeError: słownik zmienił rozmiar podczas iteracji.

Dzieje się tak, ponieważ mydict.keys()zwraca iterator, a nie listę. Jak wskazano w komentarzach, wystarczy przekonwertować mydict.keys()na listę list(mydict.keys())i powinna działać.

Prosty test w konsoli pokazuje, że nie można modyfikować słownika podczas iteracji:

>>> mydict = {'one': 1, 'two': 2, 'three': 3, 'four': 4}
>>> for k, v in mydict.iteritems():
...    if k == 'two':
...        del mydict[k]
...
------------------------------------------------------------
Traceback (most recent call last):
  File "<ipython console>", line 1, in <module>
RuntimeError: dictionary changed size during iteration

Jak stwierdzono w odpowiedzi delnana, usuwanie wpisów powoduje problemy, gdy iterator próbuje przejść do następnego wpisu. Zamiast tego użyj keys()metody, aby uzyskać listę kluczy i pracować z tym:

>>> for k in mydict.keys():
...    if k == 'two':
...        del mydict[k]
...
>>> mydict
{'four': 4, 'three': 3, 'one': 1}

Jeśli chcesz usunąć na podstawie wartości pozycji, użyj items()metody:

>>> for k, v in mydict.items():
...     if v == 3:
...         del mydict[k]
...
>>> mydict
{'four': 4, 'one': 1}

Możesz to zrobić w dwóch krokach:

remove = [k for k in mydict if k == val]
for k in remove: del mydict[k]

Moim ulubionym podejściem jest zwykle po prostu nowy dykt:

# Python 2.7 and 3.x
mydict = { k:v for k,v in mydict.items() if k!=val }
# before Python 2.7
mydict = dict((k,v) for k,v in mydict.iteritems() if k!=val)

Nie można modyfikować kolekcji podczas iteracji. W ten sposób leży szaleństwo - przede wszystkim, jeśli pozwolono by ci usunąć i usunąć bieżący element, iterator musiałby przejść dalej (+1), a następne wezwanie nextzabrałoby cię poza to (+2), więc w końcu pomijam jeden element (ten zaraz za tym, który usunąłeś). Masz dwie opcje:

• Skopiuj wszystkie klucze (lub wartości lub oba, w zależności od potrzeb), a następnie powtórz je. Możesz do tego użyć .keys()et al (w Pythonie 3 przekaż wynikowy iterator list). Może to być bardzo marnotrawne pod względem przestrzennym.
• Powtarzaj mydictjak zwykle, zapisując klucze do usunięcia w osobnej kolekcji to_delete. Po zakończeniu iteracji mydictusuń wszystkie elementy to_deletez mydict. Oszczędza trochę (w zależności od tego, ile kluczy zostało usuniętych i ile pozostało) miejsca w pierwszym podejściu, ale wymaga również kilku dodatkowych wierszy.

Zamiast tego powtarzaj kopię, taką jak ta zwrócona przez items():

for k, v in list(mydict.items()):

Najczystszy w użyciu list(mydict):

>>> mydict = {'one': 1, 'two': 2, 'three': 3, 'four': 4}
>>> for k in list(mydict):
...     if k == 'three':
...         del mydict[k]
... 
>>> mydict
{'four': 4, 'two': 2, 'one': 1}

Odpowiada to równoległej strukturze list:

>>> mylist = ['one', 'two', 'three', 'four']
>>> for k in list(mylist):                            # or mylist[:]
...     if k == 'three':
...         mylist.remove(k)
... 
>>> mylist
['one', 'two', 'four']

Oba działają w python2 i python3.

Możesz użyć słownika.

d = {k:d[k] for k in d if d[k] != val}

W przypadku python3, iteracja na dic.keys () podniesie błąd rozmiaru słownika. Możesz użyć tego alternatywnego sposobu:

Testowany z python3, działa dobrze i nie pojawia się komunikat o błędzie „ słownik zmienił rozmiar podczas iteracji ”:

my_dic = { 1:10, 2:20, 3:30 }
# Is important here to cast because ".keys()" method returns a dict_keys object.
key_list = list( my_dic.keys() )

# Iterate on the list:
for k in key_list:
    print(key_list)
    print(my_dic)
    del( my_dic[k] )


print( my_dic )
# {}

Możesz najpierw zbudować listę kluczy do usunięcia, a następnie iterować tę listę, usuwając je.

dict = {'one' : 1, 'two' : 2, 'three' : 3, 'four' : 4}
delete = []
for k,v in dict.items():
    if v%2 == 1:
        delete.append(k)
for i in delete:
    del dict[i]

Istnieje sposób, który może być odpowiedni, jeśli elementy, które chcesz usunąć, zawsze znajdują się na „początku” iteracji nagrania

while mydict:
    key, value = next(iter(mydict.items()))
    if should_delete(key, value):
       del mydict[key]
    else:
       break

„Początek” jest gwarantowany tylko w przypadku niektórych wersji / implementacji języka Python. Na przykład z Co nowego w Pythonie 3.7

natura zachowywania kolejności wstawiania obiektów dict została uznana za oficjalną część specyfikacji języka Python.

W ten sposób unika się kopii dyktu, którą sugeruje wiele innych odpowiedzi, przynajmniej w Pythonie 3.

Wypróbowałem powyższe rozwiązania w Python3, ale wydaje się, że to jedyne, które działa dla mnie podczas przechowywania obiektów w nagraniu. Zasadniczo tworzysz kopię swojego dict () i iterujesz ją, usuwając wpisy z oryginalnego słownika.

        tmpDict = realDict.copy()
        for key, value in tmpDict.items():
            if value:
                del(realDict[key])