1. 程式人生 > 程式設計 >Python使用pyyaml模組處理yaml資料

Python使用pyyaml模組處理yaml資料

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}

以上就是本文的全部內容,希望對大家的學習有所幫助,也希望大家多多支援我們。