玩转数据处理利器:学会使用 YAML 文件轻松处理数据

时间:2024-01-24 08:13:23

测试管理班是专门面向测试与质量管理人员的一门课程,通过提升从业人员的团队管理、项目管理、绩效管理、沟通管理等方面的能力,使测试管理人员可以更好的带领团队、项目以及公司获得更快的成长。提供 1v1 私教指导,BAT 级别的测试管理大咖量身打造职业规划。

玩转数据处理利器:学会使用 YAML 文件轻松处理数据_Python

YAML文件处理

什么是YAML文件

YAML全称其实是"YAML Ain't a Markup Language"(YAML不是一种标记语言)的递归缩写,所以它强调的是数据本身,而不是以标记为重点。

YAML 是一种可读性非常高,与程序语言数据结构非常接近。同时具备丰富的表达能力和可扩展性,并且易于使用的数据标记语言。

为什么要使用YAML文件

其实YAML文件也是一种配置文件,但是相较于ini,conf配置文件来说,更加的简洁,操作简单,还能存放不同类型的数据,而像ini存储的值就都是字符串类型,读取之后还要手动转换。

YAML的基本语法规则

  • 大小写敏感
  • 使用缩进表示层级关系
  • 缩进时不允许使用Tab键,只允许使用空格。(可以将你的ide的tab按键输出替换成4个空格)
  • 缩进的空格数目不重要,只要相同层级的元素左侧对齐即可
  • # 表示注释

YAML 的数据结构

  • 对象:键值对的集合,又称为映射(mapping)/ 哈希(hashes) / 字典(dictionary)
  • 数组:一组按次序排列的值,又称为序列(sequence) / 列表(list)
  • 纯量(scalars):单个的、不可再分的值
对象类型

对象的一组键值对,使用冒号结构表示,会转换成 Python 中的字典。

YAML:

animals:dog

Python:

{'animals': 'dog'}

YAML:

person:{name:Tom, age:20, gender:male}

Python:

{'person': {'name': 'Tom', 'age': 20, 'gender': 'male'}}
数组类型

数组类型使用 - 为前缀,每个元素独占一行,通过缩进关系表示层级包含关系,会转换成 Python 中的列表。

YAML:

-one-two-three-four-five

Python:

['one', 'two', 'three', 'four', 'five']

YAML:

--1-2-3--4-5-6

Python:

[[1, 2, 3], [4, 5, 6]]

纯量类型

纯量类型是最基本的、不可再分的值;类似基本数据类型。

  • 字符串, 不需要使用双引号包裹
  • 布尔值,true,True,false,False都可以
  • 整数
  • 浮点数
  • 时间,时间使用ISO 8601格式,时间和日期之间使用T连接,最后使用+代表时区
  • 日期,日期必须使用ISO 8601格式,即 yyyy-MM-dd
  • Null,~ 表示 Null

YAML:

int: 12float: 12.3string: petsbool: trueNone: nulltime: 2001-12-14t21:59:43.10-05:00date: 2018-03-21

Python:

{   'int': 12,    'float': 12.3,    'string': 'pets',    'bool': True,    'None': None,    'time': datetime.datetime(2001, 12, 14, 21, 59, 43, 100000, tzinfo=datetime.timezone(datetime.timedelta(days=-1, seconds=68400))),    'date': datetime.date(2018, 3, 21)}

复杂结构

YAML:

cool_list:  - 10  - 15  - 12
hard_list:  - {key: value}  - [1,2,3]  - test:      - 1      - 2      - 3
twice_list:  -    - {a: AA}    - {b: BB}    - {c: CC}

Python:

{    'cool_list': [10, 15, 12],    'hard_list': [        {'key': 'value'},        [1, 2, 3],        {            'test': [1, 2, 3]        }    ],    'twice_list':[        [            {'a': 'AA'},            {'b': 'BB'},            {'c': 'CC'}        ]    ]}

Tips:有 : 后面的内容就解析成字典,有 - 后面的内容就解析成列表的元素

YAML 文件处理

Python 中,可以使用第三方模块 PyYAML 来处理 YAML 文件。

安装 PyYAML 模块
pip install pyyaml
读取 YAML 文件

YAML 模块使用 safe_load() 方法读取 yaml 文件,在读取文件之前,和普通文件一样,需要先将文件打开。

import yaml
# 读取 YAML 文件, 以前面复杂结果数据为例with open('data.yaml', 'r') as file:    data = yaml.safe_load(file)
# 处理读取到的数据print(data['cool_list'])print(data['hard_list'][2]['test'])
写入 YAML 文件

YAML 模块使用 safe_dump() 方法向 yaml 文件中写入数据,在写入文件之前,也需要先将文件打开。

import yaml
# 要写入的数据data = {    'key1': 'value1',    'key2': 'value2',    'key3': {        'key4': 'value4'    }}
# 写入 YAML 文件with open('output.yaml', 'w') as file:    yaml.safe_dump(data, file)