字典排序
问题
创建字典,并在迭代或序列化字典的时候控制元素的顺序。
解决方法
Python 提供的 collections
模块中 OrderedDict
,就能够控制字典中元素的顺序。在执行迭代操作的时候,元素会保持插入时的顺序,示例代码如下:
from collections import OrderedDict
d = OrderedDict()
d['foo'] = 1
d['bar'] = 2
d['spam'] = 3
d['grok'] = 4
for key in d:
print(key, d[key])
# 输出结果
# foo 1
# bar 2
# spam 3
# grok 4
OrderedDict
在构建需要序列化或者编码成其他格式的映射同样适用。比如,控制以 JSON 编码后字段的顺序:
>>> import json
>>> json.dumps(d)
'{"foo": 1, "bar": 2, "spam": 3, "grok": 4}'
>>>
小结
其实 OrderedDict
内部维护一个根据键插入顺序排序的双向链表。当新的元素插入进来的时候,会被放到链表的尾部。而且已经存在的键重复赋值并不会改变键的顺序。
注意: 因为,OrderedDict
内部维护另外一个链表,它的大小是普通字典的两倍。所以使用的同时,需要权衡使用 OrderedDict
的好处是否要大于额外内存的消耗等影响。