collections之defaultdict

来源:互联网 发布:mfc编程实例pdf 编辑:程序博客网 时间:2024/06/01 09:06

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)





0 0
原创粉丝点击