python之pyyaml模块

pyyaml模块在python中用于处理yaml格式数据，主要使用yaml.safe_dump()、yaml.safe_load()函数将python值和yaml格式数据相互转换。固然也存在yaml.dump()、yaml.load()函数，一样能实现数据转换功能，只是官方不太推荐使用。官方给出的解释，由于yaml.safe_dump()、yaml.safe_load() 可以：

Resolve only basic YAML tags. This is known to be safe for untrusted input.

若是想对一个yaml文件中的多块yaml数据进行转换操做，则能够使用yaml.safe_dump_all()、yaml.safe_load_all()函数。下面会说明yaml.safe_dump()、yaml.safe_load()、yaml.safe_dump_all()、yaml.safe_load_all()函数的用法。

 

yaml.safe_dump()

将一个python值转换为yaml格式文件，示例以下：

import yaml dict_data = {'a': 1, 'b': 2} with open('data.yaml', 'w', encoding='UTF-8') as yaml_file: yaml.safe_dump(dict_data, yaml_file)

若是上述yaml.dump()中不带第二个参数，则会返回一个相似yaml格式的字符串

import yaml dict_data = {'a': 1, 'b': 2} yaml_string = yaml.safe_dump(dict_data) print(type(yaml_string)) print(yaml_string)

运行结果：

<class 'str'> a: 1 b: 2

yaml.safe_load()

将yaml格式文件转换为python值，接第一例子，示例以下：

import yaml with open('data.yaml', encoding='UTF-8') as yaml_file: data = yaml.safe_load(yaml_file) print(type(data)) print(data)

运行结果：

<class 'dict'> {'a': 1, 'b': 2}

yaml.safe_dump_all()

将一序列的python值转换为yaml格式文件，若是yaml.safe_dump_all()中不带第二个参数，则与yaml.dump()相似，会返回一个相似yaml格式的字符串

import yaml dict_data1 = {'a': 1, 'b': 2} dict_data2 = {'c': 3, 'd': 4} yaml_string = yaml.safe_dump_all([dict_data1, dict_data2]) print(type(yaml_string)) print(yaml_string)

运行结果：

<class 'str'> a: 1 b: 2
--- c: 3 d: 4

yaml.safe_load_all()

将yaml格式文件转换为python值，该yaml文件能够包含多块yaml数据，用法以下：

import yaml with open('data.yaml', encoding='UTF-8') as yaml_file: data = yaml.safe_load_all(yaml_file) for item in data: print(item)

运行结果：

{'a': 1, 'b': 2} {'c': 3, 'd': 4}