Python 中字典的鍵映射多個值

問題

怎么實現一個鍵對應多個值的字典（也叫 multidict ）？

 

解決方案

字典中一個鍵對應一個單值的映射。如果想一個鍵映射多個值，就需要將多個值放到另外的容器中。比如列表、集合等。

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

選擇使用列表還是集合，取決實際需求。如果想保持元素插入的順序使用列表，如果想去掉重復元素使用集合（並不關心元素的順序問題）

 

 

實際上使用 collections 模塊中的 defaultdict 來構造這樣的字典。

defaultdict 的一個特征是它會自動初始化每個 key 剛開始對應的值，所以只需要添加元素即可。如：

from collections import defaultdict

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

print(d)

defaultdict(<class 'list'>, {'a': [1, 2], 'b': [4]})

from collections import defaultdict

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

print(d)

defaultdict(<class 'set'>, {'a': {1, 2}, 'b': {4}})

 

 

注意：defaultdict 會自動為將要訪問的鍵創建映射實體（即使該鍵不存在）。如果不需要這樣，可以用 setdefault() 代替。

d = {} # 一個普通的字典
d.setdefault('a', []).append(1)
d.setdefault('a', []).append(2)
d.setdefault('b', []).append(4)

print(d)

{'a': [1, 2], 'b': [4]}

但是很多程序員覺得 setdefault() 使用起來別扭。因為每次調用都得創建一個新的初始值（例如程序中的空列表）。 

 

 

來自：python3-cookbook