字典中的鍵映射多個值

問題

怎樣實現一個鍵對應多個值的字典 (也叫 multidict )？

解法

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

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

選擇使用列表還是集合取決于你的實際需求。如果你想保持元素的插入順序就應該使用列表，如果想去掉重復元素并且不關心元素的順序問題，就使用集合。

你可以很方便的使用 collections 模塊中的 defaultdict 來構造這樣的字典。defaultdict 的一個特征是它會自動初始化每個 key 剛開始對應的值，所以你只需要關注添加元素操作了。比如：

from collections import defaultdictd = 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)

總結

以上是生活随笔為你收集整理的《Python Cookbook 3rd》笔记（1.6）：字典中的键映射多个值的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。