Python3 - 字典中的键映射多个值

问题

怎样实现一个键对应多个值的字典(也叫 multidict)?

解决方案

一个字典就是一个键对应一个单值的映射。如果想要一个键映射多个值,那么就需要将这多个值放到另外的容器中, 比如列表或者集合里面。示例:

a = {
     'c': [1, 2, 3],
     'd': [4, 5]
}

b = {
     'c': {1, 2, 3},
     'd': {4, 5}
}


选择使用列表还是集合取决于实际需求。如果想保持元素的插入顺序就应该使用列表, 如果想去掉重复元素就使用集合(并且不关心元素的顺序问题)。

使用 collections 模块中的 defaultdict 可以构造这样的字典。

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

from collections import defaultdict

a = defaultdict(list)
a['c'].append(1)
a['c'].append(2)
a['d'].append(4)

b = defaultdict(set)
b['c'].add(1)
b['c'].add(2)
b['d'].add(4)

print(a)
print(b)

defaultdict(<class 'list'>, {'c': [1, 2], 'd': [4]})
defaultdict(<class 'set'>, {'c': {1, 2}, 'd': {4}})


在一个普通的字典上,可以使用 setdefault() 方法来实现。比如:

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

print(d)

{'a': [1, 2], 'b': [4]}


讨论

一般来讲,创建一个多值映射字典是很简单的。但如果选择自己实现的话,那么对于值的初始化可能会有点麻烦, 示例:

d = {}
for key, value in pairs:
    if key not in d:
        d[key] = []
    d[key].append(value)


使用 defaultdict 的话代码更加简洁:

d = defaultdict(list)
for key, value in pairs:
    d[key].append(value)
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容

  • 写在前面的话 代码中的# > 表示的是输出结果 输入 使用input()函数 用法 注意input函数输出的均是字...
    FlyingLittlePG阅读 3,055评论 0 9
  • 1. 迭代对象解压赋值 解压赋值操作可以应用到任何迭代对象上,如:列表、元组、字符串、文件对象、迭代器、生成器。 ...
    faris_shi阅读 1,145评论 0 0
  • 本文为《爬着学Python》系列第九篇文章。 从现在开始算是要进入“真刀真枪”的Python学习了。之所以这么说,...
    SyPy阅读 2,172评论 0 14
  • 朴树,这个名字对于多数九零后而言可能会有些陌生,开始对他有认识可能是韩寒所导演的《后会无期》中的主题曲《平...
    L陌生人阅读 469评论 1 1
  • 亲爱的孩子: 你现在已经满两岁半了,妈妈计划着每隔半年给你写一封信,来记录一下你每一段时间的成长经历,因为...
    英格乐英语贾老师阅读 974评论 3 4