Python cookbook（数据结构与算法）在字典中将键映射到多个值上的方法

Python Cookbook （数据结构与算法）中提供了一种在字典中将键映射到多个值上的方法，使用标准库模块collections的defaultdict类可以轻松实现。

defaultdict

defaultdict类是一种字典的子类，它重载了一个方法并添加了一个可读实例变量。它在创建字典时需要一个工厂函数作为参数。当查询不存在的键时，会自动使用该工厂函数生成默认值。

下面是定义一个defaultdict实例的示例代码：

from collections import defaultdict

d = defaultdict(list)

上面代码指定了一个列表作为defaultdict的默认值。如果查询的键不存在，就会返回一个空列表。

append()方法

我们可以使用append()方法来添加字典中的键值对。当键不存在时，会先创建默认值列表，然后才能执行append()方法来添加值。

下面是一个例子：

from collections import defaultdict

d = defaultdict(list)
d['a'].append(1)
d['a'].append(2)
d['b'].append(3)

print(d)
# defaultdict(<class 'list'>, {'a': [1, 2], 'b': [3]})

上面代码首先创建了一个defaultdict实例，并将列表作为默认值。接着将键'a'映射到列表[1, 2]上，再将键'b'映射到列表[3]上。最后打印了整个defaultdict实例。

setdefault()方法

还可以使用setdefault()方法来向defaultdict中的列表添加元素，下面是示例代码：

from collections import defaultdict

d = defaultdict(list)
d.setdefault('a', []).append(1)
d.setdefault('a', []).append(2)
d.setdefault('b', []).append(3)

print(d)
# defaultdict(<class 'list'>, {'a': [1, 2], 'b': [3]})

上面代码首先创建了一个defaultdict实例，并为键'a'设置默认值列表。接着通过setdefault()方法，如果键不存在就创建一个默认值列表，并向列表添加元素。最后打印了整个defaultdict实例。

总结：使用defaultdict类可以快速简单的实现将键映射到多个值这种复杂的数据结构。无需担心键是否存在的问题，只需简单调用append()或setdefault()方法即可添加元素到正确的列表中。