Comment supprimer des éléments d'un dictionnaire tout en itérant dessus?

Est-il légitime de supprimer des éléments d'un dictionnaire en Python tout en itérant dessus?

Par exemple:

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

L'idée est de supprimer les éléments qui ne remplissent pas une certaine condition du dictionnaire, au lieu de créer un nouveau dictionnaire qui est un sous-ensemble de celui qui est itéré.

Est-ce une bonne solution? Existe-t-il des moyens plus élégants / efficaces?

ÉDITER:

Cette réponse ne fonctionnera pas pour Python3 et donnera un RuntimeError.

RuntimeError: la taille du dictionnaire a changé pendant l'itération.

Cela se produit car mydict.keys()renvoie un itérateur et non une liste. Comme indiqué dans les commentaires, il suffit de convertir mydict.keys()en liste list(mydict.keys())et cela devrait fonctionner.

Un simple test dans la console montre que vous ne pouvez pas modifier un dictionnaire en itérant dessus:

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

Comme indiqué dans la réponse de delnan, la suppression d'entrées provoque des problèmes lorsque l'itérateur essaie de passer à l'entrée suivante. Utilisez plutôt la keys()méthode pour obtenir une liste des clés et travaillez avec:

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

Si vous devez supprimer en fonction de la valeur des éléments, utilisez items()plutôt la méthode:

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

Vous pouvez également le faire en deux étapes:

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

Mon approche préférée est généralement de simplement faire un nouveau dict:

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

Vous ne pouvez pas modifier une collection pendant son itération. De cette façon réside la folie - plus particulièrement, si vous étiez autorisé à supprimer et à supprimer l'élément en cours, l'itérateur devrait passer à (+1) et le prochain appel à vous nextmènerait au-delà (+2), vous finissent par sauter un élément (celui juste derrière celui que vous avez supprimé). Vous avez deux options:

• Copiez toutes les clés (ou les valeurs, ou les deux, selon vos besoins), puis parcourez-les. Vous pouvez utiliser .keys()et al pour cela (en Python 3, passez l'itérateur résultant à list). Cela pourrait être très gaspillant en termes d'espace.
• Répétez mydictcomme d'habitude, en enregistrant les clés à supprimer dans une collection séparée to_delete. Une fois l'itération terminée mydict, supprimez tous les éléments to_deletede mydict. Économise de l'espace (en fonction du nombre de clés supprimées et du nombre restant) sur la première approche, mais nécessite également quelques lignes supplémentaires.

Répétez plutôt une copie, comme celle retournée par items():

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

C'est le plus propre à utiliser 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}

Cela correspond à une structure parallèle pour les listes:

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

Les deux fonctionnent en python2 et python3.

Vous pouvez utiliser une compréhension de dictionnaire.

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

Avec python3, itérer sur dic.keys () augmentera l'erreur de taille du dictionnaire. Vous pouvez utiliser cette méthode alternative:

Testé avec python3, cela fonctionne bien et l'erreur "La taille du dictionnaire a changé pendant l'itération " n'est pas déclenchée:

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

Vous pouvez d'abord créer une liste de clés à supprimer, puis parcourir cette liste en les supprimant.

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]

Il existe un moyen qui peut convenir si les éléments que vous souhaitez supprimer sont toujours au "début" de l'itération du dict

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

Le «début» n'est garanti que pour être cohérent pour certaines versions / implémentations Python. Par exemple de Quoi de neuf dans Python 3.7

la nature de préservation de l'ordre d'insertion des objets dict a été déclarée comme faisant officiellement partie de la spécification du langage Python.

De cette façon, vous évitez une copie du dict que la plupart des autres réponses suggèrent, au moins en Python 3.

J'ai essayé les solutions ci-dessus en Python3 mais celle-ci semble être la seule qui fonctionne pour moi lors du stockage d'objets dans un dict. Fondamentalement, vous faites une copie de votre dict () et vous répétez cela en supprimant les entrées de votre dictionnaire d'origine.

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