python中实现yaml文件管理参数

yaml参数管理器

这是文件目录关系，其中config存放.yaml文件，scripts存放py文件
然后就可以自由使用了：

import yaml
import os

#获取路径
script_directory = os.path.dirname(os.path.abspath(__file__))
# 相对于脚本文件的路径
image_relative_path = "../config/1.yaml"



# 构建文件的完整路径
path = os.path.join(script_directory, image_relative_path)


try:
    with open(path, 'r') as file:
        yaml_data = yaml.load(file, Loader=yaml.FullLoader)

except FileNotFoundError:
    print("ControlParameter.yaml 文件不存在")

print(yaml_data['k2'])

除了手动写入yaml外，还可以在程序中书写：

写入YAML文件的方法

def write_yaml(yaml_path,data):
    with open(yaml_path, encoding="utf-8", mode="w") as f:
        yaml.dump(data,stream=f,allow_unicode=True)

a = 10

data = {
    'name': 'John Doe',
    'age': 30,
    'pa': a,
    'city': 'New York'
}  # 字典是单引号的

write_yaml(path, data)

yaml本身就是一个字典，所以可以适当使用提高我们的效率。