Python讀取YAML的完整指南

一、安裝pyyaml

使用Python讀取YAML文件需要先安裝pyyaml庫，使用pip命令即可安裝：

pip install pyyaml

二、讀取YAML文件

在python中，可以使用pyyaml庫的load函數讀取YAML文件：

import yaml

with open('example.yaml', 'r') as file:
    content = yaml.load(file, Loader=yaml.FullLoader)

上面的代碼將example.yaml文件讀取到content變量中。需要注意的是，在使用pyyaml時，為了避免潛在的安全問題，建議使用FullLoader。

三、讀取字符串

如果要讀取YAML格式的字符串，可以使用pyyaml庫的load函數：

import yaml

content = yaml.load('name: John\nage: 30\n', Loader=yaml.FullLoader)

上面的代碼將字符串”name: John\nage: 30\n”讀取到content變量中。

四、讀取多個文檔

如果要讀取多個YAML文件或字符串，每個文件或字符串都是一個單獨的文檔，可以使用pyyaml庫的load_all函數：

import yaml

with open('example.yaml', 'r') as file:
    content = yaml.load_all(file, Loader=yaml.FullLoader)

for doc in content:
    print(doc)

上面的代碼將example.yaml文件中的多個文檔讀取到content變量中，然後使用for循環遍歷輸出。

五、讀取特定節點

如果只需要讀取YAML文件或字符串中的一個節點，可以使用pyyaml庫的safe_load函數，然後使用字典方式讀取：

import yaml

content = yaml.safe_load('name: John\nage: 30\n')
print(content['name']) # 輸出 John

上面的代碼只讀取了’name’節點。

六、寫入YAML文件

使用pyyaml庫可以將Python對象寫入YAML格式的文件中，使用dump函數即可：

import yaml

data = {
    'name': 'John',
    'age': 30
}

with open('example.yaml', 'w') as file:
    yaml.dump(data, file)

上面的代碼將Python對象data寫入example.yaml文件中。

七、寫入字符串

如果要將Python對象寫入YAML格式的字符串中，可以使用pyyaml庫的dump函數：

import yaml

data = {
    'name': 'John',
    'age': 30
}

content = yaml.dump(data)
print(content)

八、總結

本文介紹了使用Python讀取和寫入YAML文件的方法，包括安裝pyyaml庫、讀取YAML文件、讀取YAML字符串、讀取多個文檔、讀取特定節點、寫入YAML文件和寫入YAML字符串。使用這些方法，我們可以方便地讀取和寫入YAML文件和字符串。