众所周知,在Python中如果访问字典中不存在的键,会引发KeyError异常(JavaScript中如果对象中不存在某个属性,则返回undefined)。比如下面的统计单词数量的代码就会在第一次统计单词时抛出KeyError异常。
strings = ('puppy', 'kitten', 'puppy', 'puppy',
'weasel', 'puppy', 'kitten', 'puppy')
counts = {}
for kw in strings:
counts[kw] += 1
dict.setdefault()方法
上述的例子可以采用setdefault()方法解决默认值的问题,使之第一次遇到某个单词时不会抛出KeyError异常。如下:
strings = ('puppy', 'kitten', 'puppy', 'puppy',
'weasel', 'puppy', 'kitten', 'puppy')
counts = {}
for kw in strings:
counts.setdefault(kw, 0)
counts[kw] += 1
collections.defaultdict类
以上的方法虽然在一定程度上解决了dict中不存在默认值的问题,但是这时候我们会想,有没有一种字典它本身提供了默认值的功能呢?答案是肯定的,那就是collections.defaultdict。
- 其传入参数必须是type, 例如int, list,set或函数。而返回的默认值,就是这些type下的默认值,例如int返回的默认值是0,list是空list等。
- defaultdict与普通dict的最大作用在于: 你可以直接call一个不存在的key,如果不存在这个key,那就先直接创建这个key,并根据默认值的设置,赋值value,而后再继续操作。
所以上述的问题就可以这么解决:
from collections import defaultdict
strings = ('puppy', 'kitten', 'puppy', 'puppy',
'weasel', 'puppy', 'kitten', 'puppy')
counts = defaultdict(lambda: 0) # 使用lambda来定义简单的函数
for s in strings:
counts[s] += 1
参考大佬:http://code.activestate.com/recipes/523034/。