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

原文地址:https://www.cnblogs.com/keye/p/13216568.html