快捷導(dǎo)航

Python標(biāo)準(zhǔn)庫(kù)defaultdict模塊使用示例

更新時(shí)間：2015年04月28日 09:08:57 投稿：junjie

這篇文章主要介紹了Python標(biāo)準(zhǔn)庫(kù)defaultdict模塊使用示例,本文講解了如何使用defaultdict給字典value元素添加默認(rèn)類型以及defaultdict的兩個(gè)使用小案例,需要的朋友可以參考下

Python標(biāo)準(zhǔn)庫(kù)中collections對(duì)集合類型的數(shù)據(jù)結(jié)構(gòu)進(jìn)行了很多拓展操作，這些操作在我們使用集合的時(shí)候會(huì)帶來很多的便利，多看看很有好處。
defaultdict是其中一個(gè)方法，就是給字典value元素添加默認(rèn)類型，之前看到過但是沒注意怎么使用，今天特地瞅了瞅。

首先是各大文章介紹的第一個(gè)例子：

復(fù)制代碼代碼如下:

import collections as coll  

def default_factory():  

    return 'default value'  

d = coll.defaultdict(default_factory, foo='bar')  

print 'd:',d  

print 'foo=>', d['foo']  

print 'foo=>', d['bar']   #key為'bar'的元素不存在，會(huì)有一個(gè)默認(rèn)值

輸出結(jié)果是這樣的：

復(fù)制代碼代碼如下:

d: defaultdict(<function default_factory at 0x022E6E70>, {'foo': 'bar'})  

foo=> bar  

foo=> default value  

結(jié)論：這里可以看到，當(dāng)我們?nèi)∽值渲幸粋€(gè)沒有的key的時(shí)候就會(huì)自動(dòng)根據(jù) default_factory生成一個(gè)value，類似于 d.get('bar','default value')

一個(gè)對(duì)比的例子：
如果一個(gè)字典中有個(gè)map的value是一個(gè)set，往這個(gè)set連續(xù)添加兩個(gè)元素，使用原始的dict是這樣的

復(fù)制代碼代碼如下:

dict_set1 = {}  

#如果不知道這個(gè)字段中key有沒有，需要先判斷  

if 'key' not in dict_set1:  

    dict_set1['key'] = set()  

dict_set1['key'].add('111')  

dict_set1['key'].add('000')  

print dict_set1

如果使用defaultdict就是這樣的

復(fù)制代碼代碼如下:

dict_set = coll.defaultdict(set)  

dict_set['key'].add('000')  

dict_set['key'].add('111')  

print dict_set

好處就是不需要做set的初始化這個(gè)判斷了。

兩個(gè)使用的小案例

復(fù)制代碼代碼如下:

ss = '1111222233334444'
dict_int = coll.defaultdict(int)
for s in ss:
dict_int[s] += 1
print dict_int

'''''
官方文檔的這個(gè)例子就能看到這種寫法的簡(jiǎn)潔了
https://docs.python.org/2/library/collections.html#collections.defaultdict
>>> s = [('yellow', 1), ('blue', 2), ('yellow', 3), ('blue', 4), ('red', 1)]
>>> d = defaultdict(list)
>>> for k, v in s:
... d[k].append(v)
...
>>> d.items()
[('blue', [2, 4]), ('red', [1]), ('yellow', [1, 3])]
'''

當(dāng)我們做這種帶有統(tǒng)計(jì)性質(zhì)的數(shù)據(jù)操作時(shí)候，這個(gè)對(duì)象很好用。

您可能感興趣的文章: