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}