博客

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

  • Post category:Python

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

前言

在实际工作中，我们需要频繁地操作Excel表格，但直接使用Excel软件来处理可能会非常繁琐，而Python作为一种简单实用的编程语言，可以极大地简化我们的工作。在本篇文章中，我将介绍如何使用Python来操作Excel表格，并提供一些封装好的函数，以便大家更加方便地使用。

准备工作

在使用Python操作Excel之前，我们需要先安装一个开源的第三方库——openpyxl。我们可以通过pip命令来安装该库，具体操作如下：

pip install openpyxl

安装好openpyxl之后，我们就可以开始使用Python来操作Excel表格了。

主要代码实现

创建Excel表格

我们可以使用openpyxl库的Workbook对象来创建一个空的Excel表格，具体代码如下：

from openpyxl import Workbook

# 创建一个新的Excel表格对象
wb = Workbook()

# 选择当前活跃的工作表
ws = wb.active

# 在A1单元格中写入数据
ws['A1'] = 'Hello, world!'

# 保存表格
wb.save('test.xlsx')

上述代码中，我们首先创建一个新的Workbook对象，然后选择当前活跃的工作表，并向其中的A1单元格写入了一条数据。最后，我们使用save()方法将Excel表格保存至本地。

读取Excel表格

如果我们需要读取已有的Excel表格中的数据，可以使用openpyxl库的load_workbook()方法来加载该表格，然后使用Worksheet对象中的rows或columns属性来获取表格中的数据。具体代码如下：

from openpyxl import load_workbook

# 加载Excel表格
wb = load_workbook('test.xlsx')

# 获取当前活跃的工作表
ws = wb.active

# 从A1单元格中读取数据
data = ws['A1'].value

# 打印读取结果
print(data)

上述代码中，我们使用了load_workbook()方法来加载已有的Excel表格，并获取到其当前活跃的工作表。然后，我们使用ws[‘A1’].value来获取A1单元格中的数值，并将其打印出来。

封装代码为函数

为了更方便地使用操作Excel的代码，在这里我封装了两个函数来实现创建Excel表格和读取Excel表格中的数据。具体代码如下：

from openpyxl import Workbook, load_workbook

def create_excel(filename):
    """
    创建一个新的Excel表格
    :param filename: 保存的文件名
    """
    wb = Workbook()
    ws = wb.active
    ws.append(['name', 'age', 'gender'])
    ws.append(['Tom', 20, 'male'])
    ws.append(['Lucy', 18, 'female'])
    ws.append(['Jack', 22, 'male'])
    wb.save(filename)

def read_excel(filename):
    """
    读取指定Excel表格中的数据
    :param filename: Excel表格文件名
    """
    wb = load_workbook(filename)
    ws = wb.active
    for row in ws.rows:
        for cell in row:
            print(cell.value, end='\t')
        print()

上述代码中，create_excel()函数用于创建一个新的Excel表格并向其中写入数据；read_excel()函数用于读取指定Excel表格中的数据并打印出来。

示例展示

我们可以通过调用上述函数来实现操作Excel表格，具体代码如下：

# 创建一个新的Excel表格
create_excel('test.xlsx')

# 读取Excel表格中的数据
read_excel('test.xlsx')

运行上述代码后，我们可以在终端中看到如下输出结果：

name    age     gender
Tom     20      male
Lucy    18      female
Jack    22      male

总结

在本篇文章中，我们介绍了如何使用Python来操作Excel表格，并提供了两个函数用于创建Excel表格和读取其中的数据。同时，我们还学习了如何使用openpyxl库来访问Excel表格中的数据，这将大大简化我们的工作。

Tags: ,