让你的字典读取不报错

我们知道,读取 Python 字典时,如果某个 Key 不存在,Python 就会报错,如下图所示:

为了防止它报错,我们可以使用.get()方法:

但每次都要写.get()稍显麻烦。于是我们可以通过自己实现一个自定义的字典对象,来解决这个问题。

我们自定义一个类,继承dict,然后实现它的__missing__方法:

1
2
3
4
5
6
class OurDict(dict):
def __missing__(self, key):
return None
a = OurDict({'a': 1, 'b': 2})
print(a['x'])

运行效果如下图所示:

只要这个 Key 不存在,就会返回 None

在 Python 的collections工具模块中,有一个defaultdict实现了类似的事情。我们看它的使用方法。

1
2
3
4
5
6
7
8
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 不存在时,就自动以当前时间作为默认值,并存入到字典中。

谢乾坤 | Kingname wechat
第一时间获取最新文章更新,请订阅我的微信公众号:未闻Code