Python標準庫defaultdict模塊使用示例

字號:


    Python標準庫中collections對集合類型的數(shù)據(jù)結構進行了很多拓展操作,這些操作在我們使用集合的時候會帶來很多的便利,多看看很有好處。
    defaultdict是其中一個方法,就是給字典value元素添加默認類型,之前看到過但是沒注意怎么使用,今天特地瞅了瞅。
    首先是各大文章介紹的第一個例子:
    代碼如下:
    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'的元素不存在,會有一個默認值
    輸出結果是這樣的:
    代碼如下:
    d: defaultdict(, {'foo': 'bar'})
    foo=> bar
    foo=> default value
    結論:這里可以看到,當我們取字典中一個沒有的key的時候就會自動根據(jù) default_factory生成一個value,類似于 d.get('bar','default value')
    一個對比的例子:
    如果一個字典中有個map的value是一個set,往這個set連續(xù)添加兩個元素,使用原始的dict是這樣的
    代碼如下:
    dict_set1 = {}
    #如果不知道這個字段中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就是這樣的
    代碼如下:
    dict_set = coll.defaultdict(set)
    dict_set['key'].add('000')
    dict_set['key'].add('111')
    print dict_set
    好處就是不需要做set的初始化這個判斷了。
    兩個使用的小案例
    代碼如下:
    ss = '1111222233334444'
    dict_int = coll.defaultdict(int)
    for s in ss:
    dict_int[s] += 1
    print dict_int
    '''''
    官方文檔的這個例子就能看到這種寫法的簡潔了
    >>> 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])]
    '''
    當我們做這種帶有統(tǒng)計性質的數(shù)據(jù)操作時候,這個對象很好用。