Python操作Excel数据的封装函数分享

我们来详细讲解一下“Python操作Excel数据的封装函数分享”的完整实例教程。

标题

1. 概述

在实际开发中，经常需要对Excel表格进行操作，例如读写Excel表格中的数据，新增、删除、修改Excel表格中的数据等。Python是一种非常强大的编程语言，提供了许多操作Excel的库，例如：xlrd、openpyxl、xlwt等。然而，使用这些库进行操作时，我们需要写很多重复的代码，代码也很难维护。因此，在本篇文章中，我们将介绍如何封装一些常用的Python操作Excel数据的函数，使得Excel操作变得更加便捷和高效。

2. 实现环境

在实现以下函数之前，我们需要先安装一些必要的Python库，包括：

• xlrd
• xlwt
• openpyxl

3. 读取Excel数据

我们经常需要读取Excel表格中的数据，并进行一些操作。以下是一个简单的函数，用于读取Excel表格中的数据：

import xlrd

def read_excel(file_path, sheet_name):
    """
    读取Excel表格中的数据
    :param file_path: Excel文件路径
    :param sheet_name: 工作表名称
    :return: Excel中的数据列表
    """
    data_list = []  # 用于保存数据的列表
    # 打开Excel文件，获取工作簿对象
    workbook = xlrd.open_workbook(file_path)
    # 根据工作表名称获取工作表对象
    sheet = workbook.sheet_by_name(sheet_name)
    # 获取工作表中的行数和列数
    n_rows = sheet.nrows
    n_cols = sheet.ncols
    # 遍历工作表中的每一行
    for i in range(n_rows):
        row_list = []  # 用于保存一行数据的列表
        # 遍历该行中的每一列数据
        for j in range(n_cols):
            # 获取该单元格中的数据
            value = sheet.cell_value(i, j)
            row_list.append(value)
        data_list.append(row_list)
    return data_list

该函数接收两个参数，分别是Excel文件路径和工作表名称。函数返回一个二维列表，其中包含了Excel中的所有数据。以下是一个示例：

data_list = read_excel('data.xlsx', 'Sheet1')
print(data_list)

输出结果如下所示：

[['编号', '姓名', '性别', '年龄'], 
 ['001', '张三', '男', 18], 
 ['002', '李四', '女', 19], 
 ['003', '王五', '男', 20]]

4. 写入Excel数据

除了读取Excel数据，我们还需要能够向Excel表格中写入数据。以下是一个简单的函数，用于向Excel表格中写入数据：

import xlwt

def write_excel(file_path, sheet_name, data_list):
    """
    向Excel表格中写入数据
    :param file_path: Excel文件路径
    :param sheet_name: 工作表名称
    :param data_list: 数据列表
    """
    # 创建工作簿对象
    workbook = xlwt.Workbook(encoding='utf-8')
    # 创建工作表对象
    sheet = workbook.add_sheet(sheet_name)
    # 遍历数据列表，写入工作表
    for i in range(len(data_list)):
        row_list = data_list[i]  # 获取数据列表中的一行数据
        # 遍历该行数据，写入工作表
        for j in range(len(row_list)):
            value = row_list[j]
            # 向单元格中写入数据
            sheet.write(i, j, value)
    # 保存Excel文件
    workbook.save(file_path)

该函数接收三个参数，分别是Excel文件路径、工作表名称和数据列表。以下是一个示例：

data_list = [['编号', '姓名', '性别', '年龄'], 
             ['001', '张三', '男', 18], 
             ['002', '李四', '女', 19], 
             ['003', '王五', '男', 20]]
write_excel('output.xlsx', 'Sheet1', data_list)

运行以上代码后，将会在程序所在的目录下生成一个名为“output.xlsx”的Excel文件，其中包含了刚才写入的数据列表。

5. 结语

通过本篇文章介绍的Python操作Excel数据的封装函数，我们可以更加方便和高效地操作Excel表格中的数据。当然，这些函数仅仅是最基础的操作，我们还可以根据具体需求进行更加丰富的功能拓展。