Python字典:collections 中 defaultdict 的使用

1.字典的键映射多个值

将下面的列表转成字典

l = [('a',2),('b',3),('a',1),('b',4),('a',3),('a',1),('b',3)]

一个字典就是一个键对应一个单值的映射，而上面的列表中有相同键。如果你想要一个键映射多个值，那么就需要将这多个值放到另外的序列中，比如 list 或者 set 里面，像下面这样:

d = {

'a': [1, 2, 3],

'b': [4, 5]

}

e = {

'a': {1, 2, 3},

'b': {4, 5}

}

你可以很方便的使用 collections 模块中的 defaultdict 来构造这样的字典。 defaultdict 的一个特征是它会自动初始化每个 key 刚开始对应的值。

In [1]: l = [('a',2),('b',3),('a',1),('b',4),('a',3),('a',1),('b',3)]

In [2]: from collections import defaultdict

In [3]: d = defaultdict(list)

In [4]: for key, value in l:

...: d[key].append(value)

...:

In [5]: d

Out[5]: defaultdict(list, {'a': [2, 1, 3, 1], 'b': [3, 4, 3]})

当然这个默认的容器不一定是 list, 也可以是集合 set。根据自己的需求选择用 list 还是 set 。如果你想保持元素的插入顺序就应该使用列表，如果想去掉重复元素就使用集合！

2.统计字典中某个值出现的次数

来源于微信交流群里一个朋友工作中的问题，列表中有很多字典，需要统计字典中相同的键对应的值的和

利用 defaultdict 设置默认值的方法 defaultdict(int)，代码如下:

In [6]: d = defaultdict(int)

In [7]: objs = [{'F29958SVDK6': 12}, {'F29958SVDK6': 12}, {'F29958SVDK6': 12}, {'F29958SVDK6': 12}, {'F29958SVDK6': 12}]

In [8]: for obj in objs:

...: for key,value in obj.items():

...: d[key] += value

...:

In [9]: d

Out[9]: defaultdict(int, {'F29958SVDK6': 60})

二、collections 创建有序字典

字典dict是无序的，如果我们想要有序的dict，可以使用OrdereDict 。示例如下：

In [11]: from collections import OrderedDict

In [12]: d = OrderedDict()

In [13]: d['bar'] = 2

In [14]: d['non'] = 8

In [15]: d['sek'] = 5

In [16]: d

Out[17]: OrderedDict([('bar', 2), ('non', 8), ('sek', 5)])

OrderedDict 内部维护着一个根据键插入顺序排序的双向链表。每次当一个新的元素插入进来的时候，它会被放到链表的尾部。对于一个已经存在的键的重复赋值不会改变键的顺序。

需要注意的是，一个 OrderedDict 的大小是一个普通字典的两倍，因为它内部维护着另外一个链表。所以如果你要构建一个需要大量 OrderedDict 实例的数据结构的时候 (比如读取 100,000 行 CSV 数据到一个 OrderedDict 列表中去)，那么你就得仔细权衡一下是否使用 OrderedDict 带来的好处要大过额外内存消耗的影响。