简单介绍Python中如何给字典设置默认值

共计 1299 个字符，预计需要花费 4 分钟才能阅读完成。

我们都知道，在 Python 的字典里边，如果 key 不存在的话，通过 key 去取值是会报错的。

>>> aa = {'a':1, 'b':2}
>>> aa['c']
Traceback (most recent call last):
  File "", line 1, in 
KeyError: 'c'

如果我们在取不到值的时候不报错而是给定一个默认值的话就友好多了。

>>> from collections import defaultdict
>>> aa = defaultdict(int)
>>> aa['a'] = 1
>>> aa['b'] = 2
>>> aa
defaultdict(, {'a': 1, 'b': 2})
>>> aa['c']
0
>>> aa
defaultdict(, {'a': 1, 'b': 2, 'c': 0})
>>> aa = dict.fromkeys('abc', 0)
>>> aa
{'a': 0, 'b': 0, 'c': 0}

defaultdict(default_factory) 中的 default_factory 也可以传入自定义的匿名函数之类的哟。

>>> aa = defaultdict(lambda : 1)
>>> aa['a']
1

这里有个比较特殊的点：只要对应的 key 已经被设定了值之后，那么对相同 key 再次设置默认值就没用了。

因此，如果你在循环里边给一个 key 重复设定默认值的话，那么也只会第一次设置的生效。

>>> aa = {'a':1, 'b':2}
>>> aa
{'a': 1, 'b': 2}
>>> aa.get('c')
>>> aa.setdefault('c', 'hello')
'hello'
>>> aa.get('c')
'hello'
>>> aa
{'a': 1, 'b': 2, 'c': 'hello'}
>>> aa.setdefault('c', 'world')
'hello'
>>> aa.get('c')
'hello'

>>> aa = {'a':1, 'b':2}
>>> aa
{'a': 1, 'b': 2}
>>> aa['c']
Traceback (most recent call last):
  File "", line 1, in 
KeyError: 'c'
>>> aa.get('c')
>>> aa
{'a': 1, 'b': 2}
>>> aa.get('c', 'hello')
'hello'
>>> aa.get('b')
2

防止 keyerror 创建带默认值的字典

from collections import defaultdict
data = collections.defaultdict(lambda :[])

阿里云 2 核 2G 服务器 3M 带宽 61 元 1 年，有高配

腾讯云新客低至 82 元 / 年，老客户 99 元 / 年

代金券：在阿里云专用满减优惠券