Python-defaultdict(type)字典

众所周知,在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/。

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值