Python操作Excel文件：掌握openpyxl模块

翻滚的胖子
2023-05-25 13:48:17
后端技术
Python Java

Python操作Excel文件：掌握openpyxl模块

一、openpyxl模块介绍

openpyxl是Python中一个用于操作Excel文件的第三方库，它可以读取、写入Excel文件，并且支持Excel文件的修改、格式化等操作。使用openpyxl模块可以大大提高Python处理Excel文件的效率。下面我们来了解一下openpyxl模块的基本使用方法。

二、安装openpyxl模块

在使用openpyxl模块之前，需要先安装openpyxl模块。安装openpyxl模块可以使用pip命令：

pip install openpyxl

安装完成后，我们就可以使用openpyxl模块了。

三、打开Excel文件

使用openpyxl模块打开Excel文件，可以使用load_workbook函数，该函数可以接收Excel文件的路径，并返回一个Workbook对象。

from openpyxl import load_workbook

workbook = load_workbook('example.xlsx')

上面的代码中，我们使用load_workbook函数打开了example.xlsx文件，并将返回的Workbook对象赋值给了workbook变量。

四、读取Excel文件中的数据

在openpyxl模块中，我们可以通过Worksheet对象来访问Excel文件中的单元格。使用Worksheet对象可以获取单元格的值、样式、格式等信息。下面我们来看一下如何读取Excel文件中的数据。

1. 获取Worksheet对象

要获取Worksheet对象，可以使用Workbook对象的active属性，该属性可以返回第一个Worksheet对象。

worksheet = workbook.active

上面的代码中，我们使用workbook.active属性获取了第一个Worksheet对象，并将其赋值给了worksheet变量。

2. 读取单元格的值

使用Worksheet对象可以获取单元格的值，可以使用cell函数，该函数接收行号和列号作为参数，并返回该单元格的值。

cell_value = worksheet.cell(row=1, column=1).value

上面的代码中，我们使用worksheet.cell函数获取第一行第一列单元格的值，并将其赋值给了cell_value变量。

3. 读取一行或一列的数据

使用Worksheet对象可以获取一行或一列的数据，可以使用rows或columns属性，这两个属性分别返回一行或一列的数据。

# 获取第一行数据
for cell in worksheet[1]:
    print(cell.value)

# 获取第一列数据
for cell in worksheet['A']:
    print(cell.value)

上面的代码中，我们使用worksheet[1]获取第一行数据，使用worksheet['A']获取第一列数据，并分别打印出了每个单元格的值。

五、写入Excel文件

在openpyxl模块中，我们可以通过Worksheet对象来写入Excel文件。使用Worksheet对象可以设置单元格的值、样式、格式等属性。下面我们来看一下如何写入Excel文件。

1. 设置单元格的值

使用Worksheet对象可以设置单元格的值，可以使用cell函数，该函数接收行号和列号作为参数，并可以使用value属性设置单元格的值。

worksheet.cell(row=1, column=1).value = 'Python'

上面的代码中，我们使用worksheet.cell函数设置第一行第一列单元格的值为'Python'。

2. 设置单元格的样式

使用openpyxl模块可以设置单元格的样式，例如字体、大小、颜色等。可以使用Font、Alignment、Border等类来设置单元格的样式。

from openpyxl.styles import Font, Alignment, Border, Side

# 设置字体
font = Font(name='Arial', size=12, bold=True)

# 设置对齐方式
alignment = Alignment(horizontal='center', vertical='center')

# 设置边框
border = Border(left=Side(border_style='thin', color='000000'), 
                right=Side(border_style='thin', color='000000'),
                top=Side(border_style='thin', color='000000'), 
                bottom=Side(border_style='thin', color='000000'))

# 设置单元格样式
worksheet.cell(row=1, column=1).font = font
worksheet.cell(row=1, column=1).alignment = alignment
worksheet.cell(row=1, column=1).border = border

上面的代码中，我们设置了第一行第一列单元格的字体、对齐方式和边框。