亲宝软件园·资讯

展开

Python YAML文件

alwaysrun 人气:0

YAML是一种数据序列化格式,方便人类阅读,且容易和脚本语言交互。常用于配置文件,也用于数据存储或传输。

YAML格式

YAML三种基本数据类型:

1.标量:如字符串、整数和浮点数、日期

2.数组:使用前导符号-表示,且符号与值间要有空格;

3.对象:键值对集合;用冒号+空格: 分割;

YAML文件

文件基本规则:

区分大小写;

使用缩进表示层级关系:

字符串不需要使用引号括起,但有特殊字符串(如空格、冒号等)时必须使用引号;

使用#注释;

一个文件中可包含多份YAML文档:

---(即三个中划线)表示一份文档的开始;

[可选]用...(即三个小数点)表示一份文档的结束;

---
# 这是第一份文档
one: 1
# 其他内容...
...

---
# 这是第二份文档
two: 2
# 其他内容...

YAML操作

Python中有两个YAML模块:PyYAML和ruamel.yaml(pip3 install pyyamlpip3 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
  }
]

加载全部内容

相关教程
猜你喜欢
用户评论