python中创建和修改yaml文件的方法

import os
import yaml
desired_caps = {
                'train': 'dataTrain/2007_train.txt',
                'val': 'dataTrain/2007_val.txt',
                'nc': 2,
                'names': ['a','b']
                }
curpath = os.path.dirname(os.path.realpath(__file__))
yamlpath = os.path.join(curpath, "./yamlFile/caps.yaml")
# 写入到yaml文件
with open(yamlpath, "w", encoding="utf-8") as f:
    yaml.dump(desired_caps, f)

import os
import yaml
def set_state(state):
    file_name = "./yamlFile/bottlemldel.yaml"
    with open(file_name) as f:
        doc = yaml.safe_load(f)
    doc['nc'] = state
    with open(file_name, 'w') as f:
        yaml.safe_dump(doc, f, default_flow_style=False)
set_state(8)

pip install ruamel.yaml

from ruamel.yaml import YAML
yaml = YAML()
# 读取yaml文件
with open("environment.yaml", "r", encoding='utf-8') as file:
    data = yaml.load(file)
# 修改yaml文件
datas = data["dependencies"]
for index in range(len(datas)):
    curStr = datas[index]
    # 删除最后一个等号及其后面的内容
    if isinstance(curStr, str) and curStr.count("=") > 1:
        # 从右边开始以"="为分界分割一次
        strs = datas[index].rsplit("=", 1)
        datas[index] = strs[0]
        print(datas[index])
# 保存yaml文件
with open("environment.yaml", "w", encoding='utf-8') as file:
    yaml.dump(data, file)