在Python上进行迭代时从字典中删除项目是否合法?
例如:
for k, v in mydict.iteritems(): if k == val: del mydict[k]
这个想法是从字典中删除不满足特定条件的元素,而不是创建一个新字典,该字典是正在迭代的字典的子集。
这是一个好的解决方案吗?有没有更优雅/更有效的方法?
此答案不适用于Python3,并且会给出RuntimeError。
Python3
RuntimeError
RuntimeError:字典在迭代过程中更改了大小。
发生这种情况是因为mydict.keys()返回的是迭代器而不是列表。正如评论中指出的那样,只需将其转换mydict.keys()为列表即可list(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}