Python collections.defaultdict 筆記

其實defaultdict 就是一個字典，只不過python自動的為它的鍵賦了一個初始值。這也就是說，你不顯示的為字典的鍵賦初值python不會報錯，看下實際例子。

比如你想計算頻率

frequencies = {}
for word in wordlist:
    frequencies[word] += 1

python會拋出一個KeyError 異常，因為字典索引之前必須初始化，可以用下面的方法解決

for word in wordlist:
    try:
        frequencies[word] += 1
    except KeyError:
        frequencies[word] = 1

for word in wordlist:
    if word in frequencies:
        frequencies[word] += 1
    else:
        frequencies[word] = 1

當然，collections.defaultdict也可以輕松的解決這個問題

from collections import defaultdict
frequencies = defaultdict(int) #傳入int()函數來初始化
for word in wordlist:
    frequencies[word] += 1

collections.defaultdict可以接受一個函數作為參數來初始化。什么意思呢，看上面的例子，我們想要frequencies[word]初始化為0，這時就可以用一個int()函數作為參數出給defaultdict，我們不帶參數調用int()，int()就會返回一個0值