Python字典中的鍵對映多個值(列表或者集合)
阿新 • • 發佈:2018-11-10
一個字典就是一個鍵對應一個單值的對映。如果你想要一個鍵對映多個值,那麼你就需要將這多個值放到另外的容器中, 比如列表或者集合裡面。比如,你可以像下面這樣構造這樣的字典:
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)
d = defaultdict(set)
d['a'].add(1)
d['a'].add(2)
d['b'].add(4)
需要注意的是, defaultdict 會自動為將要訪問的鍵(就算目前字典中並不存在這樣的鍵)建立對映實體。 如果你並不需要這樣的特性,你可以在一個普通的字典上使用 setdefault() 方法來代替。比如:
d = {} # A regular dictionary
d.setdefault('a', []).append(1)
d.setdefault('a', []).append(2)
d.setdefault('b', []).append(4)
但是很多程式設計師覺得 setdefault() 用起來有點彆扭。因為每次呼叫都得建立一個新的初始值的例項(例子程式中的空列表 [] )。
討論
一般來講,建立一個多值對映字典是很簡單的。但是,如果你選擇自己實現的話,那麼對於值的初始化可能會有點麻煩, 你可能會像下面這樣來實現:
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)