如何在Python中保留YAML字典的遍历顺序?
哈哈,这个问题我之前踩过坑!YAML里的默认映射(就是咱们写的键值对)在不少解析器里都不保证顺序,你遇到的thing2跑前面的情况太常见了,给你几个实用的解决办法:
1. 用YAML 1.2原生的有序映射类型
YAML 1.2专门提供了!!omap类型来定义有序映射,写法很简单:
!!omap - thing1: value1 - thing2: value2
只要你的YAML解析库支持这个类型(比如PyYAML的最新版本),解析后会得到一个有序的数据结构(比如Python里的OrderedDict),遍历的时候就严格按照你写的顺序来。
2. 给解析器指定有序加载逻辑(以PyYAML为例)
如果不想改YAML文件的写法,可以自定义加载器,让解析出来的字典保持顺序。比如在Python里这么写:
import yaml from collections import OrderedDict def ordered_yaml_load(stream): class OrderedLoader(yaml.Loader): pass def construct_ordered_mapping(loader, node): loader.flatten_mapping(node) return OrderedDict(loader.construct_pairs(node)) OrderedLoader.add_constructor( yaml.resolver.BaseResolver.DEFAULT_MAPPING_TAG, construct_ordered_mapping ) return yaml.load(stream, OrderedLoader)
用这个ordered_yaml_load函数去加载你的YAML内容,得到的就是有序的OrderedDict,遍历顺序和你写的完全一致。
3. 改用列表嵌套结构(兼容性最强)
要是担心不同解析器对!!omap的支持不一致,或者不想折腾加载逻辑,直接换个YAML结构就行——用列表来包裹每个需要排序的项:
things: - key: thing1 value: value1 - key: thing2 value: value2
列表的顺序在所有YAML解析器里都是严格保留的,遍历的时候直接循环这个列表,逻辑清晰还不会出顺序问题,我自己在需要严格顺序的场景里经常用这个方案。
另外提一句:Python 3.7及以上的普通字典本身是保序的,但有些旧版本的YAML库可能还在用无序的实现,所以如果你的环境是新Python,升级PyYAML到最新版,默认加载可能就会保序,但上面的方法还是更稳妥。
内容的提问来源于stack exchange,提问作者LCM
相关产品推荐
相关产品推荐

