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

Question

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

例如：

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

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

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

Answer 1

编辑：

对于 Python3（或更高版本）：

>>> mydict
{'four': 4, 'three': 3, 'one': 1}

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

其余答案适用于 Python2 但不适用于 Python3 并引发RuntimeError。

RuntimeError：字典在迭代期间改变了大小。

这是因为mydict.keys() 返回的是迭代器而不是列表。 正如 cmets 中所指出的，只需将 mydict.keys() 转换为 list(mydict.keys()) 的列表，它应该可以工作。

---

对于python2：

控制台中的一个简单测试表明您无法在迭代字典时对其进行修改：

>>> 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 值删除，请改用items() 方法：

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

Answer 2

您也可以分两步完成：

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

我最喜欢的方法通常是制作一个新的字典：

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

Answer 3

改为遍历一个副本，例如 items() 返回的那个：

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

Answer 4

您不能在迭代集合时对其进行修改。这种方式很疯狂 - 最值得注意的是，如果您被允许删除并删除当前项目，则迭代器将不得不继续（+1）并且下一次调用 next 将带您超越（+2），所以你最终会跳过一个元素（你删除的元素后面的那个元素）。你有两个选择：

• 复制所有键（或值，或两者，取决于您的需要），然后遍历它们。您可以为此使用 .keys() 等（在 Python 3 中，将生成的迭代器传递给 list）。不过在空间方面可能会非常浪费。
• 像往常一样遍历mydict，将要删除的键保存在单独的集合to_delete 中。完成对 mydict 的迭代后，从 mydict 中删除 to_delete 中的所有项目。与第一种方法相比，节省了一些（取决于删除了多少键以及保留了多少键）空间，但也需要多几行。

Answer 5

使用 python3，迭代 dic.keys() 会引发字典大小错误。您可以使用这种替代方式：

用python3测试，效果很好，没有出现错误“dictionary changed size during iteration”：

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

Answer 6

使用 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}

这对应于列表的并行结构：

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

在 python2 和 python3 中都可以工作。

Answer 7

您可以使用字典理解。

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

Answer 8

您可以先构建一个要删除的键列表，然后遍历该列表以删除它们。

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]

Answer 9

如果您要删除的项目始终位于 dict 迭代的“开始”，则有一种方法可能适合

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

“开始”仅保证对于某些 Python 版本/实现是一致的。例如来自What’s New In Python 3.7

dict 对象的插入顺序保存特性已被宣布为 Python 语言规范的官方部分。

这种方式避免了许多其他答案建议的 dict 副本，至少在 Python 3 中是这样。

Answer 10

我在 Python3 中尝试了上述解决方案，但在将对象存储在 dict 中时，这似乎是唯一对我有用的解决方案。基本上，您制作 dict() 的副本并在删除原始字典中的条目时对其进行迭代。

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