python excel和yaml文件的读取封装

当我们在使用Python处理数据的时候，经常会遇到读取Excel表格、YAML格式文件的需求。为了提高代码重用性和可维护性，我们可以对这些操作进行封装，使得代码更加简洁易懂。

在Python中，我们可以使用openpyxl库来读取Excel文件，PyYAML库来读取YAML文件，下面我们通过示例来具体介绍如何实现这些功能。

1. Python处理Excel文件

首先，我们需要安装openpyxl库，可以使用pip命令进行安装：

pip install openpyxl

接下来，我们来看一个读取Excel文件的例子：

import openpyxl

def read_excel_file(file_path):
    # 打开文件
    wb = openpyxl.load_workbook(filename=file_path)

    # 获取所有工作表名字
    sheet_names = wb.sheetnames

    # 读取第一个工作表
    ws = wb[sheet_names[0]]

    # 获取行数、列数
    row_num = ws.max_row
    col_num = ws.max_column

    # 读取数据
    data = []
    for i in range(1, row_num+1):
        row_data = []
        for j in range(1, col_num+1):
            cell_value = ws.cell(row=i, column=j).value
            row_data.append(cell_value)
        data.append(row_data)

    return data

在上面的代码中，我们首先定义了一个函数read_excel_file，该函数接收一个Excel文件路径参数file_path，然后使用load_workbook方法加载Excel文件。接下来，我们获取所有工作表的名称，然后获取第一个工作表。我们通过max_row和max_column方法获取该工作表的行数和列数。我们使用循环读取每一行每一列的数据，然后将其以二维列表的形式返回。

我们可以使用下面的代码来调用该函数：

data = read_excel_file("example.xlsx")
print(data)

运行上面的代码，可以得到Excel文件中的所有数据。

2. Python处理YAML文件

YAML是一种轻量级、可读性高的标记语言，常用于配置文件。在Python中，我们可以使用PyYAML库来读取YAML文件。首先，我们需要安装该库，可以使用pip命令进行安装：

pip install pyyaml

然后，我们来看一个读取YAML文件的例子：

import yaml

def read_yaml_file(file_path):
    with open(file_path, "r") as f:
        data = yaml.load(f, Loader=yaml.FullLoader)
    return data

在上面的代码中，我们首先定义了一个函数read_yaml_file，该函数接收一个YAML文件路径参数file_path，然后使用open方法打开文件，并使用yaml.load方法将其转换为Python对象。注意，在使用yaml.load方法时，需要指定Loader参数为yaml.FullLoader，以避免潜在的安全问题。

我们可以使用下面的代码来调用该函数：

data = read_yaml_file("example.yaml")
print(data)

运行上面的代码，可以得到YAML文件中的所有数据。

至此，我们已经介绍了如何封装Python读取Excel文件和YAML文件的功能。通过这样的封装，在后续的开发中，我们可以直接调用这些函数，而不需要每次都写重复的代码。