软件测试|使用Python读写yaml文件，你会了吗？

简介

YAML（YAML Ain't Markup Language）是一种可读的数据序列化格式，它常用于配置文件和数据交换。Python 提供了许多库来处理 YAML 文件。在本文中，我们将探讨如何使用 PyYAML 库来读取和写入 YAML 文件，以及提供一些示例来说明其用法。

环境准备

在读写yaml文件之前，我们需要先安装好环境，我们需要先安装 PyYAML 库，安装命令如下：

pip install PyYAML

读取yaml文件

使用 PyYAML 可以轻松读取 YAML 文件。以下是一个读取 YAML 文件的示例：

import yaml

# 打开并读取 YAML 文件
with open('config.yaml', 'r') as file:
    data = yaml.load(file, Loader=yaml.FullLoader)

# 访问 YAML 数据
print(data)

在上述示例中，我们首先使用 open() 函数打开 YAML 文件，然后使用 yaml.load() 函数加载文件中的数据。 Loader=yaml.FullLoader 参数用于指定加载器。最后，我们访问加载的 YAML 数据并打印它。

写入 YAML 文件

要将数据写入 YAML 文件，你可以使用 PyYAML 的 yaml.dump() 函数。以下是一个写入 YAML 文件的示例：

import yaml

# 创建要写入的数据
data = {
    'name': 'Alice',
    'age': 30,
    'city': 'Wonderland'
}

# 打开并写入 YAML 文件
with open('output.yaml', 'w') as file:
    yaml.dump(data, file)

在上述示例中，我们首先创建一个包含数据的字典。然后，使用 open() 函数打开一个输出文件，并使用 yaml.dump() 函数将数据写入文件中。

高级示例

通常，YAML 文件中包含更复杂的数据结构，如列表和嵌套字典。以下是一个包含更复杂数据的示例：

import yaml

data = {
    'name': 'Bob',
    'languages': ['Python', 'JavaScript', 'Java'],
    'address': {
        'street': '123 Main St',
        'city': 'Anytown',
        'zip': '12345'
    }
}

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

加载复杂的 YAML 文件

要加载包含复杂数据结构的 YAML 文件，你可以使用相同的 yaml.load() 函数，它会将 YAML 数据转换为 Python 字典。例如：

import yaml

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

print(loaded_data)

总结

Python 中读取和写入 YAML 文件是一项常见任务，特别是在处理配置文件和数据交换时。PyYAML 是一个方便的库，可以轻松地实现这些操作。希望本文提供的示例和解释有助于大家掌握如何读取和写入 YAML 文件。

最后感谢每一个认真阅读我文章的人，礼尚往来总是要有的，虽然不是什么很值钱的东西，如果你用得到的话可以直接拿走：

这些资料，对于【软件测试】的朋友来说应该是最全面最完整的备战仓库，这个仓库也陪伴上万个测试工程师们走过最艰难的路程，希望也能帮助到你！