快捷導航

python中創(chuàng)建和修改yaml文件的方法

更新時間：2023年11月01日 09:37:41 作者：程序員阿明

YAML 是 "YAML Ain’t a Markup Language"的遞歸縮寫,yaml簡潔美觀,是一種常用的標記語言,可以用來表達多種數(shù)據(jù)結(jié)構(gòu)和配置文件,本文給大家介紹python中如何創(chuàng)建和修改yaml文件,感興趣的朋友一起看看吧

python中如何創(chuàng)建和修改yaml文件

1、創(chuàng)建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")
# 寫入到y(tǒng)aml文件
with open(yamlpath, "w", encoding="utf-8") as f:
    yaml.dump(desired_caps, f)

2、修改yaml文件

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)

補充：

python如何修改yaml文件

YAML簡介

YAML 是 "YAML Ain’t a Markup Language"的遞歸縮寫。開發(fā)的這種語言時其意思其實是：“Yet Another Markup Language”。yaml簡潔美觀，是一種常用的標記語言，可以用來表達多種數(shù)據(jù)結(jié)構(gòu)和配置文件。

那么如何創(chuàng)建和修改yaml語言呢

YAML修改

1.首先需要安裝工具包ruamel.yaml

pip install ruamel.yaml

2.然后讀取yaml文件進行修改。

下面給出一個例子：對dependencies中的每一個字符串成員，刪除最后一個等號及其后面的內(nèi)容

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]
    # 刪除最后一個等號及其后面的內(nèi)容
    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)

修改前：