問題

怎樣實(shí)現(xiàn)一個(gè)鍵對(duì)應(yīng)多個(gè)值的字典(也叫 multidict )？

解決方案

一個(gè)字典就是一個(gè)鍵對(duì)應(yīng)一個(gè)單值的映射。如果你想要一個(gè)鍵映射多個(gè)值，那么你就需要將這多個(gè)值放到另外的容器中，比如列表或者集合里面。比如，你可以像下面這樣構(gòu)造這樣的字典：

d = {
    'a' : [1, 2, 3],
    'b' : [4, 5]
}
e = {
    'a' : {1, 2, 3},
    'b' : {4, 5}
}

選擇使用列表還是集合取決于你的實(shí)際需求。如果你想保持元素的插入順序就應(yīng)該使用列表，如果想去掉重復(fù)元素就使用集合（并且不關(guān)心元素的順序問題）。

你可以很方便的使用collections模塊中的defaultdict來構(gòu)造這樣的字典。defaultdict的一個(gè)特征是它會(huì)自動(dòng)初始化每個(gè)key剛開始對(duì)應(yīng)的值，所以你只需要關(guān)注添加元素操作了。比如：

from collections import defaultdict

d = defaultdict(list)
d['a'].append(1)
d['a'].append(2)
d['b'].append(4)

d = defaultdict(set)
d['a'].add(1)
d['a'].add(2)
d['b'].add(4)

需要注意的是，defaultdict會(huì)自動(dòng)為將要訪問的鍵(就算目前字典中并不存在這樣的鍵)創(chuàng)建映射實(shí)體。如果你并不需要這樣的特性，你可以在一個(gè)普通的字典上使用setdefault()方法來代替。比如：

d = {} # A regular dictionary
d.setdefault('a', []).append(1)
d.setdefault('a', []).append(2)
d.setdefault('b', []).append(4)

但是很多程序員覺得setdefault()用起來有點(diǎn)別扭。因?yàn)槊看握{(diào)用都得創(chuàng)建一個(gè)新的初始值的實(shí)例(例子程序中的空列表[])。

討論

一般來講，創(chuàng)建一個(gè)多值映射字典是很簡單的。但是，如果你選擇自己實(shí)現(xiàn)的話，那么對(duì)于值的初始化可能會(huì)有點(diǎn)麻煩，你可能會(huì)像下面這樣來實(shí)現(xiàn)：

d = {}
for key, value in pairs:
    if key not in d:
        d[key] = []
    d[key].append(value)

如果使用defaultdict的話代碼就更加簡潔了：

d = defaultdict(list)
for key, value in pairs:
    d[key].append(value)

這一小節(jié)所討論的問題跟數(shù)據(jù)處理中的記錄歸類問題有大的關(guān)聯(lián)?？梢詤⒖?.15小節(jié)的例子。