Python标准库defaultdict模块使用示例
Python标准库中collections对集合类型的数据结构进行了很多拓展操作,这些操作在我们使用集合的时候会带来很多的便利,多看看很有好处。
defaultdict是其中一个方法,就是给字典value元素添加默认类型,之前看到过但是没注意怎么使用,今天特地瞅了瞅。
首先是各大文章介绍的第一个例子:
importcollectionsascoll defdefault_factory(): return'defaultvalue' d=coll.defaultdict(default_factory,foo='bar') print'd:',d print'foo=>',d['foo'] print'foo=>',d['bar'] #key为'bar'的元素不存在,会有一个默认值
输出结果是这样的:
d:defaultdict(<functiondefault_factoryat0x022E6E70>,{'foo':'bar'}) foo=>bar foo=>defaultvalue
结论:这里可以看到,当我们取字典中一个没有的key的时候就会自动根据default_factory生成一个value,类似于d.get('bar','defaultvalue')
一个对比的例子:
如果一个字典中有个map的value是一个set,往这个set连续添加两个元素,使用原始的dict是这样的
dict_set1={} #如果不知道这个字段中key有没有,需要先判断 if'key'notindict_set1: dict_set1['key']=set() dict_set1['key'].add('111') dict_set1['key'].add('000') printdict_set1
如果使用defaultdict就是这样的
dict_set=coll.defaultdict(set) dict_set['key'].add('000') dict_set['key'].add('111') printdict_set
好处就是不需要做set的初始化这个判断了。
两个使用的小案例
ss='1111222233334444' dict_int=coll.defaultdict(int) forsinss: dict_int[s]+=1 printdict_int ''''' 官方文档的这个例子就能看到这种写法的简洁了 https://docs.python.org/2/library/collections.html#collections.defaultdict >>>s=[('yellow',1),('blue',2),('yellow',3),('blue',4),('red',1)] >>>d=defaultdict(list) >>>fork,vins: ... d[k].append(v) ... >>>d.items() [('blue',[2,4]),('red',[1]),('yellow',[1,3])] '''
当我们做这种带有统计性质的数据操作时候,这个对象很好用。