python字典dict,在使用过程中,插入元素或者更新某个值时,在不知道key是否存在的情况下,一般都会有一个if的判断 。偶然发现 collections 模块中有 defaultdict 这个类,使用很方便。defaultdict 是dict的subclass。
直接看官网上介绍的示例:
s = [('yellow', 1), ('blue', 2), ('yellow', 3), ('blue', 4), ('red', 1)]
d = defaultdict(list)
for k, v in s:
d[k].append(v)
统计词频:
s = 'mississippi'
d = defaultdict(int)
d = defaultdict(lambda:0)
for k in s:
d[k] += 1
其实我就是在看这个代码(How to Write a Spelling Corrector,21行代码实现)时发现这个defaultdict的。
另外,dict利用setdefault函数可以类似的实现这个功能,但是没有defaultdict简单和快捷。
s = [('yellow', 1), ('blue', 2), ('yellow', 3), ('blue', 4), ('red', 1)]
d = {}
for k, v in s:
d.setdefault(k, []).append(v)