Python使用pyyaml模塊處理yaml數(shù)據(jù)的方法

這篇文章主要講解了Python使用pyyaml模塊處理yaml數(shù)據(jù)的方法，內(nèi)容清晰明了，對此有興趣的小伙伴可以學(xué)習(xí)一下，相信大家閱讀完之后會有幫助。

pyyaml模塊在python中用于處理yaml格式數(shù)據(jù)，主要使用yaml.safe_dump()、yaml.safe_load()函數(shù)將python值和yaml格式數(shù)據(jù)相互轉(zhuǎn)換。當(dāng)然也存在yaml.dump()、yaml.load()函數(shù)，同樣能實現(xiàn)數(shù)據(jù)轉(zhuǎn)換功能，只是官方不太推薦使用。官方給出的解釋，因為yaml.safe_dump()、yaml.safe_load() 能夠：

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

如果想對一個yaml文件中的多塊yaml數(shù)據(jù)進行轉(zhuǎn)換操作，則可以使用yaml.safe_dump_all()、yaml.safe_load_all()函數(shù)。下面會說明yaml.safe_dump()、yaml.safe_load()、yaml.safe_dump_all()、yaml.safe_load_all()函數(shù)的用法。

yaml.safe_dump()

將一個python值轉(zhuǎn)換為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()中不帶第二個參數(shù)，則會返回一個類似yaml格式的字符串

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

運行結(jié)果：

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

yaml.safe_load()

將yaml格式文件轉(zhuǎn)換為python值，接第一例子，示例如下：

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

運行結(jié)果：

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

yaml.safe_dump_all()

將一序列的python值轉(zhuǎn)換為yaml格式文件，如果yaml.safe_dump_all()中不帶第二個參數(shù)，則與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)

運行結(jié)果：

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

yaml.safe_load_all()

將yaml格式文件轉(zhuǎn)換為python值，該yaml文件可以包含多塊yaml數(shù)據(jù)，用法如下：

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)

運行結(jié)果：

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

看完上述內(nèi)容，是不是對Python使用pyyaml模塊處理yaml數(shù)據(jù)的方法有進一步的了解，如果還想學(xué)習(xí)更多內(nèi)容，歡迎關(guān)注億速云行業(yè)資訊頻道。