如何从Python的OrderedDict中提取前N个键值对的子集?
实现合理性分析
- 你现有的代码功能上是完全可用的:逻辑直白易懂,依赖OrderedDict的有序特性,能正确提取前4个键值对,不会出现顺序错乱的问题
- 存在可优化的点:遍历了整个OrderedDict做索引判断,当字典量级很大时会产生不必要的性能损耗,同时代码写法不够简洁Pythonic
更优实现方案
通用最优方案(兼容所有Python版本,性能最优)
用itertools.islice直接截断迭代器,不会遍历超出需要的元素,性能最高,代码最短:
from itertools import islice # 提取前4个元素,返回普通dict subdict = dict(islice(my_ordered_dict.items(), 4)) # 如果需要返回的子集仍是OrderedDict类型,改成如下写法 from collections import OrderedDict sub_ordered_dict = OrderedDict(islice(my_ordered_dict.items(), 4))
小字典快速写法(适合元素量少的场景)
如果你确定字典元素不多,可以直接把items转列表后切片,写法更直观:
subdict = dict(list(my_ordered_dict.items())[:4])
注意:Python 3.7及以上版本中普通dict默认自带插入有序特性,上述方法同样适用于普通dict的前N个元素提取需求。
内容的提问来源于stack exchange,提问作者marlon
相关产品推荐
相关产品推荐

