溫馨提示×

python yaml模塊怎么使用

小億
91
2023-11-23 16:42:00
欄目: 編程語言

要使用Python的yaml模塊，首先需要安裝pyyaml庫。你可以使用以下命令來安裝它：

pip install pyyaml

安裝完成后，你可以使用以下代碼來讀取和寫入YAML文件：

import yaml

# 讀取YAML文件
with open('example.yaml', 'r') as file:
    data = yaml.safe_load(file)

# 修改數(shù)據(jù)
data['key'] = 'new value'

# 寫入YAML文件
with open('example.yaml', 'w') as file:
    yaml.dump(data, file)

在上面的代碼中，首先使用yaml.safe_load()函數(shù)來讀取YAML文件中的數(shù)據(jù)。然后，你可以通過修改data字典來修改數(shù)據(jù)。最后，使用yaml.dump()函數(shù)將修改后的數(shù)據(jù)寫入到Y(jié)AML文件中。

請注意，safe_load()和dump()函數(shù)是安全的，默認(rèn)情況下它們只允許基本的Python數(shù)據(jù)類型。如果你需要支持自定義對象的序列化和反序列化，可以使用yaml.load()和yaml.dump()函數(shù)。但是請謹(jǐn)慎使用，因為它們可能存在安全風(fēng)險。

此外，還可以使用yaml.load_all()和yaml.dump_all()函數(shù)來處理包含多個YAML文檔的文件。load_all()函數(shù)返回一個生成器，可以迭代獲取每個文檔的數(shù)據(jù)，而dump_all()函數(shù)接受一個可迭代對象來寫入多個文檔。

這只是yaml模塊的基本用法，你可以參考官方文檔以獲取更多詳細(xì)信息和更高級的用法：https://pyyaml.org/wiki/PyYAMLDocumentation

0 踩