Python编程：yaml文件读写

PyYAML is a YAML parser and emitter for Python.

译文：PyYAML是Python的YAML解析器和发射器。

文档

• http://pyyaml.org/

安装

pip install pyyaml

读取

import yaml

f = open("data.yaml", "r")
data = yaml.load(f)
f.close()

print(data)
"""
[
    {
        'name': 'PyYAML', 
        'status': 4, 
    },
    {
        'name': 'PySyck', 
        'status': 5, 
    }
]
"""

附：要读取的yaml文件

- name: PyYAML  # 备注
  status: 4
- name: PySyck
  status: 5

写入

import yaml

new_date = {
    "name": "Tom",
    "age": 23,
    "sex": "man"
}

f = open("data1.yaml", "w")
yaml.dump(new_date, f)
f.close()

"""
{age: 23, pesonname: Tom, sex: man}
"""

如果要显示结构，可以增加一个参数

#写入
d={
    'person2': 
        {
        'name':'Jack',
        'age':20, 
        'sex':'women'
        }, 
    'person1': 
        {
        'name':'Tom',
        'age':29
        }
    }

f=open("newtest.yaml","w")
yaml.dump(d, f, default_flow_style=False)
f.close() 
"""
person1:
  age: 29
  name: Tom
person2:
  age: 20
  name: Jack
  sex: women
"""

关闭警告

YAMLLoadWarning: calling yaml.load() without Loader=... is deprecated

方式一

yaml.load(input, Loader=yaml.FullLoader)

方式二

yaml.warnings({'YAMLLoadWarning': False})

方式三

PYTHONWARNINGS=ignore::yaml.YAMLLoadWarning

具体可参考
https://github.com/yaml/pyyaml/wiki/PyYAML-yaml.load(input)-Deprecation

工具方法

常用于读取配置文件

# -*- coding: utf-8 -*-
"""
yaml_util.py
"""
import yaml


def read_yaml_file(filename):
    with open(filename, "rb") as f:
        return yaml.safe_load(f)