1.6 字典中的键映射多个值

1.6 字典中的键映射多个值

问题描述

怎样实现一个键对应多个值的字典(也叫multidict)?

解决方案

可以将多个值放到另一个容器里,比如列表或集合中。选择列表或集合取决于你的实际需求,如果想保持元素的插入顺序就选择列表,如果想去掉重复元素就选择集合。
另外,你可以很方便的使用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(<class "list">, {"a": [1, 2], "b": [4]})
"""

d = defaultdict(set)
d["a"].add(1)
d["a"].add(2)
d["b"].add(4)
"""
d = defaultdict(<class "set">, {"a": {1, 2}, "b": {4}})
"""

defaultdict还有一个特性是会自动为将要访问的键(就算目前字典中不存在这个键)创建映射实体——比如上述代码若访问d["c"]则会返回一个空列表或空集合。如若你不需要这种特性,可以在一个普通字典上使用setdefault()方法来代替。比如:

d = {}
d.setdefault("a", []).append(1)
d.setdefault("a", []).append(2)
d.setdefault("b", []).append(3)
"""
d = {"a": [1, 2], "b": [3]}
"""