Python读写YAML文件全攻略

Python中操作YAML文件常用PyYAML库实现。1. 安装方法为执行pip install pyyaml；2. 读取使用yaml.safe_load()函数加载文件，注意处理编码、路径和语法错误；3. 写入使用yaml.dump()函数保存数据，需设置allow_unicode=True、sort_keys=False等参数控制输出格式；4. 处理复杂结构时应逐层访问并判断字段是否存在，结合异常处理可提升代码健壮性。掌握安装、读取、写入及结构处理技巧后即可高效操作YAML配置文件。

在Python中操作YAML文件其实挺常见的，特别是在处理配置文件的时候。YAML的结构清晰、可读性强，很多项目用它来代替JSON或者INI格式的配置。要实现读写YAML文件，最常用的是PyYAML库，下面我从实际使用角度讲讲怎么操作。

安装PyYAML库

Python标准库不包含YAML的支持，所以首先要安装第三方库：

pip install pyyaml

这个库支持基本的数据类型如字典、列表、字符串、布尔值等的转换，基本上能满足大多数配置场景的需求。

读取YAML文件

读取YAML文件是使用最多的功能之一。比如你有一个名为config.yaml的配置文件，内容如下：

database:
  host: localhost
  port: 5432
  user: admin
  password: secret
logging:
  level: debug
  file: /var/log/app.log

你可以用以下代码读取它：

import yaml

with open('config.yaml', 'r', encoding='utf-8') as f:
    config = yaml.safe_load(f)

print(config['database']['host'])  # 输出: localhost

• 使用 safe_load() 是推荐的方式，可以避免执行任意代码的风险。
• 如果你不确定文件是否存在，最好加个异常处理，比如捕获 FileNotFoundError 或 YAMLError。

常见问题：

• 编码错误：记得加上 encoding='utf-8'
• 文件路径错误：建议用绝对路径或确保相对路径正确
• YAML语法错误：缩进不对会导致解析失败

写入YAML文件

有时候你也需要把Python中的数据保存为YAML格式。比如你想生成一个默认配置：

data = {
    'app': {
        'name': 'my_app',
        'version': '1.0'
    },
    'features': ['login', 'export', 'notifications']
}

with open('output.yaml', 'w', encoding='utf-8') as f:
    yaml.dump(data, f, allow_unicode=True, sort_keys=False)

这样会生成一个类似下面的YAML文件：

app:
  name: my_app
  version: '1.0'
features:
- login
- export
- notifications

注意几个参数：

• allow_unicode=True：支持中文等非ASCII字符
• sort_keys=False：保持键的顺序，否则默认会按字母排序

如果你希望输出更美观，还可以加上 default_flow_style=False，让列表显示成换行形式。

处理复杂数据结构的小技巧

虽然大部分时候都是处理字典和列表，但有时也会遇到嵌套较深的情况。比如：

users:
  - name: Alice
    roles: [admin, user]
  - name: Bob
    roles: [user]

对应的Python结构是一个字典嵌套列表再嵌套字典。访问时要注意层级：

for user in config['users']:
    print(user['name'], user['roles'])

这时候容易出错的地方在于字段是否缺失。建议在访问前做判断：

if 'roles' in user:
    ...

或者用 .get() 方法设置默认值：

roles = user.get('roles', [])

基本上就这些了。读写YAML文件本身不难，但容易忽略细节，比如缩进、编码、异常处理等。只要掌握好这几个关键点，用起来还是很顺手的。