学习《Python Cookbook》第三版
一个字典就是一个键对应一个单值的映射。如果需要一个键映射多个值,那么就需要将多个值放到另外的容器中,比如列表或者集合里面,如:
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)
e = defaultdict(set)
e['a'].add(1)
e['b'].add(2)
e['b'].add(3)
print(d) # defaultdict(<class 'list'>, {'a': [1, 2], 'b': [4]})
print(e) # defaultdict(<class 'set'>, {'a': {1}, 'b': {2, 3}})
print("e['fd']: {}".format(e['fd'])) # e['fd']: set()
print(e.get('c')) # None
f = {}
f.setdefault('a', []).append(1)
f.setdefault('a', []).append(2)
f.setdefault('b', []).append(3)
print(f) # {'a': [1, 2], 'b': [3]}
print(f.get('fd')) # None
print(f['g']) # 报错
defaultdict会自动为将要访问的键(就算目前字典中不存在这个键)创建映射实体。
Traceback (most recent call last):
defaultdict(<class 'list'>, {'a': [1, 2], 'b': [4]})
File "E:/code/PythonCookBook/Chapter_1/py20200104.py", line 87, in <module>
print(f['g'])
KeyError: 'g'
defaultdict(<class 'set'>, {'a': {1}, 'b': {2, 3}})
e['fd']: set()
None
{'a': [1, 2], 'b': [3]}
None
如果不需要这样的特性,你可以在一个普通的字典上使用setdefault()方法替代。
虽然有很多程序员觉得使用setdefault()方法别扭。因为每次都会创建一个新的初始化的实例(程序中的空列表)。
其中,dict.get()方法不会报错,而dict[]取值如果不存在对应的键值,则抛出异常KeyError
创建一个多值映射字典很简单。但如果选择自己实现的话,对于值的初始化可能会有点麻烦,实现方法:
pairs = [
('dog', 12),
('cat', 13),
('monkey', 14),
]
animals = {}
for key, value in pairs:
if key not in animals:
animals[key] = []
animals[key].append(value)
print(animals) # {'dog': [12], 'cat': [13], 'monkey': [14]}
如果使用defautdict的话代码就更加简洁
from collections import defaultdict
pairs = [
('dog', 12),
('cat', 13),
('monkey', 14),
]
animals = defaultdict(list)
for key, value in pairs:
animals[key].append(value)
print(animals) # defaultdict(<class 'list'>, {'dog': [12], 'cat': [13], 'monkey': [14]})