如何连续多次重置defaultdict中的键且避免KeyError？

问题场景 #

当使用defaultdict时，重复删除同一个不存在的键会触发KeyError：

from collections import defaultdict
d = defaultdict(str)
d['one'] = 'hi'
del d['one']
del d['one']  # 第二次执行抛出KeyError

同样，调用d.pop('one')时，如果键不存在也会出现相同的报错问题。

我们希望用更简洁的方式实现“重置键到默认值”的效果，而不是用这种冗余的判断写法：

if 'one' in d:
    del d['one']

简洁解决方案 #

1. 利用pop的默认参数

dict.pop()方法支持传入第二个参数作为键不存在时的返回值，这样即使键不存在也不会触发异常，同时完成删除操作（存在则删除，不存在无动作）：

d.pop('one', None)

之后访问d['one']时，会自动返回defaultdict的默认值（这里是空字符串），达到重置的目的。

2. 直接赋值默认值

如果不需要删除键，只是想将对应的值重置为默认值，可以直接调用default_factory生成默认值并赋值：

d['one'] = d.default_factory()

对于defaultdict(str)来说，d.default_factory()等价于str()，会返回空字符串，直接覆盖原有值，无需判断键是否存在。

3. 封装安全删除工具函数

如果需要频繁进行这类操作，可以封装一个简单的工具函数：

from collections import defaultdict

def safe_delete(dct, key):
    if key in dct:
        del dct[key]

# 使用示例
d = defaultdict(str)
safe_delete(d, 'one')  # 无论键是否存在都不会报错

内容的提问来源于stack exchange，提问作者user3310334