yaml.load（）函數

python中讀取yaml配置文件

讀取yaml文件數據

python通過open方式讀取文件數據，再通過load函數將數據轉化為列表或字典；

import yaml
import os


def get_yaml_data(yaml_file):
    # 打開yaml文件
    print("***獲取yaml文件數據***")
    file = open(yaml_file, 'r', encoding="utf-8")
    file_data = file.read()
    file.close()
    
    print(file_data)
    print("類型：", type(file_data))


    # 將字符串轉化為字典或列表
    print("***轉化yaml數據為字典或列表***")
    data = yaml.load(file_data)
    print(data)
    print("類型：", type(data))
    return data
current_path = os.path.abspath(".")
yaml_path = os.path.join(current_path, "config.yaml")
get_yaml_data(yaml_path)

輸出：
"""
***獲取yaml文件數據***
# yaml鍵值對：即python中字典
usr: my
psw: 123455
類型：<class 'str'>
***轉化yaml數據為字典或列表***
{'usr': 'my', 'psw': 123455}
類型：<class 'dict'>
"""

 

 

yaml文件數據為鍵值對

（1）yaml文件中內容為鍵值對：

# yaml鍵值對：即python中字典
usr: my
psw: 123455
s: " abc\n"

python解析yaml文件后獲取的數據：

{'usr': 'my', 'psw': 123455, 's': ' abc\n'}

2）yaml文件中內容為“鍵值對'嵌套"鍵值對"

 

# yaml鍵值對嵌套：即python中字典嵌套字典
usr1:
  name: a
  psw: 123
usr2:
  name: b
  psw: 456

 

python解析yaml文件后獲取的數據：

 

{'usr1': {'name': 'a', 'psw': 123}, 'usr2': {'name': 'b', 'psw': 456}}

 （3）yaml文件中“鍵值對”中嵌套“數組”

# yaml鍵值對中嵌套數組
usr3:
  - a
  - b
  - c
usr4:
  - b

4. yaml文件數據為數組

（1）yaml文件中內容為數組

# yaml數組
- a
- b
- 5

python解析yaml文件后獲取的數據：

['a', 'b', 5]

（2）yaml文件“數組”中嵌套“鍵值對”

# yaml"數組"中嵌套"鍵值對"- usr1: aaa
- psw1: 111
  usr2: bbb
  psw2: 222

python解析yaml文件后獲取的數據

[{'usr1': 'aaa'}, {'psw1': 111, 'usr2': 'bbb', 'psw2': 222}]

 

 

 

load函數中的Loader=yaml.FullLoader

例子：

#讀取clinical的yaml文件
def read_clin_file(clinfile):
    with open(clinfile,'r') as CLIN:
        #備注：yaml版本5.1之后棄用，YAMLLoadWarning: calling yaml.load() without Loader=... is deprecated
        clin_info = yaml.load(CLIN,Loader=yaml.FullLoader) #加上Loader=yaml.FullLoader 避免警告。
        name = clin_info['name']
        pro_id = clin_info['sample_id']
        return(name,pro_id)