Python YAML檔案的讀寫操作詳解

2022-08-17 18:02:50

YAML格式

YAML三種基本資料型別：

1.標量：如字串、整數和浮點數、日期

布林值：“true”、“True”、“TRUE”、“yes”、“Yes"和"YES”，“false”、“False”、“FALSE”、“no”、“No"和"NO”
空：null、Null、~或不指定值；

2.陣列：使用前導符號-表示，且符號與值間要有空格；

3.物件：鍵值對集合；用冒號+空格: 分割；

YAML檔案

檔案基本規則：

區分大小寫；

使用縮排表示層級關係：

縮排要用空格（非Tab）；
相同層級元素要左對齊；

字串不需要使用引號括起，但有特殊字串（如空格、冒號等）時必須使用引號；

使用#註釋；

一個檔案中可包含多份YAML檔案：

用---（即三個中劃線）表示一份檔案的開始；

[可選]用...（即三個小數點）表示一份檔案的結束；

---
# 這是第一份檔案
one: 1
# 其他內容...
...

---
# 這是第二份檔案
two: 2
# 其他內容...

YAML操作

Python中有兩個YAML模組：PyYAML和ruamel.yaml（pip3 install pyyaml，pip3 install ruamel.yaml）。

讀取

通過load函數可方便載入檔案內容，返回的內容可能是字典、列表或空：

import yaml
def readYaml(file):  # -> Union[dict, list, None]:
    with open(file, 'r', encoding='utf-8') as f:
        return yaml.load(f, yaml.Loader)

若檔案中有多個檔案，需要使用load_all來載入；此時返回的是一個生成器，需要一次獲取每個檔案：

def readAllDoc(file) -> list:
    with open(file, 'r', encoding='utf-8') as f:
        docs = yaml.load_all(f, yaml.Loader)
        # return [d for d in docs]
        return list(docs)

儲存

yaml.dump()方法不會將列表或字典資料進行轉化yaml標準模式。需要通過ruamel.yaml來生成標準的yaml檔案。

import ruamel.yaml
def createYaml(file, data: dict):
    with open(file, 'w', encoding='utf-8') as f:
        writer = ruamel.yaml.YAML()
        writer.indent(mapping=2, sequence=4, offset=2)
        writer.dump(data, f)

通過dump_all可把陣列中的每個元素序列化為一個檔案：

def createAllDoc(file, data: list):
    with open(file, 'w', encoding='utf-8') as f:
        writer = ruamel.yaml.YAML()
        writer.indent(mapping=2, sequence=4, offset=2)
        writer.dump_all(data, f)

把[{'name': 'Mike', 'age': 12}, {'name': 'luce', 'age': 15}]序列化。

dump會生成如下包含陣列元素的yaml檔案：

  - name: Mike
    age: 12
  - name: luce
    age: 15

dump_all會生成如下yaml檔案（包含兩個檔案）：

name: Mike
age: 12
---
name: luce
age: 15

範例

yaml讀取後為列表或字典。

轉字典

讀取的內容為一個字典：

name: mike
age: 25
score: 80.5

返回的值為：

{
"name": "mike",
"age": 25,
"score": 80.5
}

獲取裡面值，先判斷是否存在，再讀取：

if 'name' in data:
  print(data['name'])

轉列表

讀取內容為一個列表：

- name: mike
  age: 24
- name: tom
  age: 30

返回的值為：

[
{
"name": "mike",
"age": 24
},
{
"name": "tom",
"age": 30
}
]

到此這篇關於Python YAML檔案的讀寫操作詳解的文章就介紹到這了,更多相關Python YAML檔案內容請搜尋it145.com以前的文章或繼續瀏覽下面的相關文章希望大家以後多多支援it145.com！

Python YAML檔案的讀寫操作詳解

目錄

YAML格式

YAML檔案

YAML操作

讀取

儲存

範例

轉字典

轉列表

熱門文章