问题描述
怎样实现一个键对应多个值的字典(也叫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]}
"""