1.6 字典中的键映射多个值

zltzlt

字典中的键映射多个值

问题

怎样实现一个键对应多个值的字典（也叫 multidict）？

解决方案

一个字典就是一个键对应一个单值的映射。

如果你想要一个键映射多个值，那么你就需要将这多个值放到另外的容器中，比如列表或者集合里面。

比如，你可以像下面这样构造这样的字典：

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

复制代码

选择使用列表还是集合取决于你的实际需求。

如果你想保持元素的插入顺序就应该使用列表，如果想去掉重复元素就使用集合（并且不关心元素的顺序问题）。

你可以很方便的使用 collections 模块中的 defaultdict 来构造这样的字典。

defaultdict 的一个特征是它会自动初始化每个 key 刚开始对应的值，所以你只需要关注添加元素操作了。

比如：

1. from collections import defaultdict
   
2. 
   
3. d = defaultdict(list)
   
4. d['a'].append(1)
   
5. d['a'].append(2)
   
6. d['b'].append(4)
   
7. 
   
8. d = defaultdict(set)
   
9. d['a'].add(1)
   
10. d['a'].add(2)
    
11. d['b'].add(4)

复制代码

需要注意的是，defaultdict 会自动为将要访问的键（就算目前字典中并不存在这样的键）创建映射实体。

如果你并不需要这样的特性，你可以在一个普通的字典上使用 setdefault() 方法来代替。比如：

1. 
   
2. d = {} # 一个普通的字典
   
3. d.setdefault('a', []).append(1)
   
4. d.setdefault('a', []).append(2)
   
5. d.setdefault('b', []).append(4)

复制代码

但是很多程序员觉得 setdefault() 用起来有点别扭。

因为每次调用都得创建一个新的初始值的实例（例子程序中的空列表 [] ）。

讨论

一般来讲，创建一个多值映射字典是很简单的。

但是，如果你选择自己实现的话，那么对于值的初始化可能会有点麻烦，你可能会像下面这样来实现：

1. d = {}
   
2. for key, value in pairs:
   
3.     if key not in d:
   
4.         d[key] = []
   
5.     d[key].append(value)

复制代码

如果使用 defaultdict 的话代码就更加简洁了：

1. d = defaultdict(list)
   
2. for key, value in pairs:
   
3.     d[key].append(value)

复制代码

594glj

谢谢大佬