软件测试|Python openpyxl库使用指南

发布时间 2023-11-15 17:27:19作者: 霍格沃兹测试开发学社

简介

我们之前介绍过,python在自动化办公方面可以大放异彩,因为Python有许多的第三方库,其中有很多库就支持我们对office软件进行操作,熟练的使用Python对office进行操作,可以实现自动化办公,极大提升我们的工作效率。本篇文章,我们就来介绍一下处理Excel的第三方库,openpyxl的使用。

安装openpyxl库

在我们使用openpyxl库之前,我们需要先安装它,安装的方式与其他的第三方库一致,使用pip命令安装即可,命令如下:

pip install openpyxl

打开Excel文件

首先,让我们看看如何打开一个已经存在的Excel文件:

import openpyxl

# 打开Excel文件
workbook = openpyxl.load_workbook('demo.xlsx')

# 选择工作表
sheet = workbook.active  # 或者使用 workbook['Sheet1']

读取和写入单元格数据

我们可以使用cell对象来读取和写入单元格的数据:

value = sheet['A1'].value
print(value)

# 写入单元格数据
sheet['B2'] = '20'
value2 = sheet['B2'].value
print(value2)

workbook.save('demo.xlsx')

------------------
输出结果如下:
姓名
20

遍历工作表数据

要遍历整个工作表的数据,您可以使用iter_rows()iter_cols()方法:

# 遍历行
for row in sheet.iter_rows(min_row=2, max_row=6, min_col=1, max_col=3):
    for cell in row:
        print(cell.value, end='\t')
    print()
--------------
输出结果如下:
Tony	20	理发师	
Linda	22	会计	
Harry	24	销售	
Muller	23	工程师	
John	30	教师

添加和删除工作表

我们还可以添加新的工作表,并删除已存在的工作表:

# 添加工作表
new_sheet = workbook.create_sheet('NewSheet')

# 删除工作表
del workbook['Sheet2']

保存Excel文件

在完成对Excel文件的操作后,不要忘记保存更改:

# 保存更改
workbook.save('demo.xlsx')

样式和格式设置

openpyxl还支持对单元格的样式和格式进行设置。我们可以对字体、背景色、对齐方式等进行设置:

我们先看一下当前demo.xlsx的样式以及单元格格式,如下图:

现在我们对样式以及单元格格式进行设置。代码如下:

import openpyxl
from openpyxl.styles import Font, PatternFill, Alignment

workbook = openpyxl.load_workbook('demo.xlsx')

# 选择工作表
sheet = workbook.active

# 设置字体
font = Font(name='Arial', bold=True)
sheet['A1'].font = font

# 设置背景色
fill = PatternFill(fill_type='solid', fgColor='FFFF00')
sheet['B1'].fill = fill

# 设置对齐方式
align = Alignment(horizontal='center', vertical='center')
sheet['C1'].alignment = align



workbook.save('demo.xlsx')

运行代码,打开demo.xlsx,如下图:

总结

openpyxl是一个功能强大的库,能够帮助我们处理和操作Excel文件。在这篇文章中,我们介绍了如何使用openpyxl库打开、读取、写入和保存Excel文件,以及如何遍历数据、添加和删除工作表,以及设置单元格样式。这只是openpyxl的一部分功能,可以在其官方文档中找到更多详细信息和用法示例。

使用openpyxl库,我们可以轻松地在Python中处理Excel文件,从而进行数据处理、报告生成等任务,而不需要依赖Microsoft Excel软件。

获取更多技术资料,请点击!