摄影:产品经理厨师:kingname
我们知道,读取 Python 字典时,如果某个 Key 不存在,Python 就会报错,如下图所示:
为了防止它报错,我们可以使用.get()
方法:
但每次都要写.get()
稍显麻烦。于是我们可以通过自己实现一个自定义的字典对象,来解决这个问题。
我们自定义一个类,继承dict
,然后实现它的__missing__
方法:
class OurDict(dict): def __missing__(self, key): returnNone a = OurDict({'a': 1, 'b': 2}) print(a['x'])
运行效果如下图所示:
只要这个 Key 不存在,就会返回 None
。
在 Python 的collections
工具模块中,有一个defaultdict
实现了类似的事情。我们看它的使用方法。
from collections import defaultdict a = defaultdict(str) a['a'] = 1 a['b'] = 2 print(a) print(a['x']) print(a)
a = defaultdict(str)
接收一个函数,让a
变量成为一个defaultdict
类型的对象,它可以像普通字典一样被赋值和读取。
但是当你读取一个不存在的 key 时,它会自动生成一个默认值,并把默认值和这个 key 插入到字典中。这个默认值取决于你在defaultdict
传入的参数。如果传入的是str
那么默认值就是空字符串,如果是int
那么默认值就是0.当然你也可以传入自定义的一个函数,从而控制这个默认值。例如我自定义一个函数,用于返回当前的时间,运行效果如下图所示:
当插入的 key 不存在时,就自动以当前时间作为默认值,并存入到字典中。