Python有意思的defaultdict方法

本文主要参考最下面的网页，简单做个提炼。

defaultdict主要解决在创建的dict为空时的赋值问题。

（1）dict.setdefault()方法

当不用defaultdict方法时，可用dict.setdefault()方法替代

下面使用参考资料中的例子举例。例子是用来统计数组中每个词出现的次数。setdefault(kw,0)方法可在dict中没有kw这个key的时候，将这个key的value设置为0,并且当dict中存在kw时，则返回其对应的value。

strings = ('puppy', 'kitten', 'puppy', 'puppy', 'weasel', 'puppy', 'kitten', 'puppy')
counts = {}
for kw in strings:
    counts[kw] = counts.setdefault(kw, 0) + 1

（2）defaultdict方法

defaultdict()方法可以接受一个类型作为参数，比如int, list等，也可以接受一个不带参数的函数作为参数。

举例1：以int作为参数，统计单词出现次数

from collections import defaultdict

strings = ('puppy', 'kitten', 'puppy', 'puppy',
           'weasel', 'puppy', 'kitten', 'puppy')
counts = defaultdict(int)  

for s in strings:
    counts[s] += 1

print(counts) # 输出是defaultdict(<class 'int'>, {'puppy': 5, 'kitten': 2, 'weasel': 1})

举例2：以一个无参数函数作为参数，同样是统计单词出现的次数

from collections import defaultdict

strings = ('puppy', 'kitten', 'puppy', 'puppy',
           'weasel', 'puppy', 'kitten', 'puppy')

def zero():
    return 0

counts = defaultdict(zero)  # zero是函数名

for s in strings:
    counts[s] += 1

print(counts) # 输出是defaultdict(<function zero at 0x0000000002062EA0>, {'puppy': 5, 'kitten': 2, 'weasel': 1})

参考资料：

http://kodango.com/understand-defaultdict-in-python