Python-在迭代字典时如何从字典中删除项目？

Question

在Python上进行迭代时从字典中删除项目是否合法？

例如：

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

这个想法是从字典中删除不满足特定条件的元素，而不是创建一个新字典，该字典是正在迭代的字典的子集。

这是一个好的解决方案吗？有没有更优雅/更有效的方法？

Answer 1

此答案不适用于Python3，并且会给出RuntimeError。

RuntimeError：字典在迭代过程中更改了大小。

发生这种情况是因为mydict.keys()返回的是迭代器而不是列表。正如评论中指出的那样，只需将其转换mydict.keys()为列表即可list(mydict.keys())，它应该可以工作。

控制台中的一个简单测试显示，在迭代字典时你无法修改字典：

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

如delnan的回答所述，当迭代器尝试移至下一个条目时，删除条目会导致问题。而是使用keys()方法获取键列表并进行处理：

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

如果需要根据项目值删除，请使用items()方法：

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