【python接口自动化】- openpyxl读取excel数据

简介:

【python接口自动化】- openpyxl读取excel数据

前言:目前我们进行测试时用于存储测试数据的软件几乎都是excel,excel方便存储和管理数据,读取数据时也比较清晰,测试时我们需要从excel从读取测试数据,结束后还需把测试结果写入到excel中,读取这一动作以往都是通过人工来操作,学会使用openpyxl后,就可以实现自动化地读取数据,解放双手。

1|0openpyxl简单介绍

​ openpyxl是一个开源项目,它是一个用于读取/写入Excel 2010文档(如xlsx 、xlsm 、xltx 、xltm文件 )的Python库,如果要处理更早格式的Excel文档(xls),需要用到其它库(如:xlrd、xlwt等),这是openpyxl比较其他模块的不足之处。openpyxl是一款比较综合的工具,不仅能够同时读取和修改Excel文档,而且可以对Excel文件内单元格进行详细设置,包括单元格样式等内容,甚至还支持图表插入、打印设置等内容。

​ python中与excel操作相关的模块:

📔 xlrd库:从excel中读取数据,支持xls、xlsx格式
📔 xlwt库:对excel进行修改操作,但不支持对xlsx格式的修改
📔 xlutils库:在xlwt和xlrd中,对一个已存在文件进行修改
📘 openpyxl库:主要针对xlsx格式的excel进行读取和编辑
2|0openpyxl的安装

pip install openpyxl

3|0openpyxl的三个基本概念

WorkBook:工作簿对象
Sheet:表单对象
Cell:表格对象
​ 也就是我们使用excel时可以看到的三个对象,我们操作excel的步骤是打开工作簿,选中表单,对单元格进行操作,而openpyxl的工作原理也是如此。

3|1openpyxl对excel进行操作

3|2新建excel

​ 运行以下代码后,就会发现项目中多了一个excel文件。

import openpyxl # 创建一个工作簿 workbook = openpyxl.Workbook() # 创建一个表单 sheet = workbook.create_sheet('表单1') # 写入一个数据 sheet.cell(row=1, column=1, value="python") # 保存 workbook.save('test.xlsx')

​ 🎈 注意点:如果保存的文件名已存在,将会覆盖原来的文件

3|3打开已存在的文件

打开已存在的文件:load_workbook()
获取表单:workbook['表单名']
读取指定表格:cell(row, column).value
import openpyxl # 打开工作簿 workbook = openpyxl.load_workbook('test.xlsx') # 获取表单 sheet = workbook['表单1'] # 读取指定的单元格数据 cell = sheet.cell(row=1, column=1).value print(cell)

​ 运行结果:

C:softwarepythonpython.exe D:/learn/python24/python_base/test.py python Process finished with exit code 0

3|4读取单个单元格

方式一:读取A6单元格的值 cell1 = sheet['A6'].value # 方式二:读取第3行,第4列单元格的值 cell2 = sheet.cell(row=3, column=4).value

3|5读取多个单元格

读取A1-B4的单元格,共8个单元格 cell3 = sheet['A1':'B4'] # 读取A1-B4的单元格,共8个单元格 cell4 = sheet['A1:B4'] # 读取第2行的单元格 cell5 = sheet[2] # 读取第1-2行的单元格 cell5 = sheet[1:2]

​ 读取多个单元格时,都是以元组的形式返回

3|6其他相关操作

关键代码 相关描述
cell(row, column, value) 写入数据,三个参数分别是行,列,值
cell(row, column).value 获取指定单元格的值,两个参数分别是行,列
workbook.remove('表单名') 删除表单
del workbook['表单名'] 删除表单
workbook.save('文件名') 保存工作簿
sheet.max_row 获取表单数据的总行数
sheet.max_column 获取表单数据的总列数
sheet.rows 获取按行所有的数据
sheet.columns 获取按列所有的数据
​ 🎈 注意点:

openpyxl中列和行的起始标识都是1,不是从0开始
从excel中读取出来的数据只有两种类型,即数值类型和字符串类型
不要随便在表格中敲空格,会影响判断最大行数和最大列数
运行操作excel的代码时,要先关闭在操作系统中打开的相关excel表,否则可能会无法读取/写入数据
写入数据时要执行保存
import openpyxl # 打开工作簿 wb = openpyxl.load_workbook('test.xlsx') # 获取表单 sh = wb['表单1'] # 读取指定的单元格数据 res1 = sh.cell(row=1, column=1).value print(res1) # 获取最大行数 print(sh.max_row) # 获取最大列数 print(sh.max_column) # 按列读取所有数据,每一列的单元格放入一个元组中 print(sh.columns) # 直接打印,打印结果是一个可迭代对象,我们可以转换成列表来查看 # 按行读取所有数据,每一行的单元格放入一个元组中 rows = sh.rows # print(list(rows)) # 转换成列表之后打印结果为具体的单元格,如下 # [(, , , , , , ), # (, , , , , , ), # (, , , , , , )] # 我们可以通过for循环以及value来查看单元格的值 for row in list(rows): # 遍历每行数据 case = [] # 用于存放一行数据 for c in row: # 把每行的每个单元格的值取出来,存放到case里 case.append(c.value) print(case)

​ 运行结果:

C:softwarepythonpython.exe D:/learn/python24/python_base/test.py case_id 3 7 ['case_id', 'interface', 'title', 'method', 'url', 'data', 'expected'] [1, 'login', '登录成功', 'post', '/member/login', '{"mobile_phone":"15692004245","pwd":"miki12345"}', "{'code': 0, 'msg': 'OK'}"] [2, 'login', '登录失败-手机号为空', 'post', '/member/login', '{"mobile_phone":"","pwd":"miki12345"}', "{'code': 2, 'msg': '手机号码为空'}"]

​ excel表中的原数据如下图:

4|0openpyxl封装

​ 在测试过程中,我们会经常需要访问excel读取测试数据,如果毎访问一次我们就调用一次openpyxl,进行打开excel,选取表单,读取数据等操作,将会增加我们的工作量和代码量。凡是需要重复使用的,我们都可以对它进行二次封装,写成我们自己的版本,还可以在封装过程中加入我们想要的内容,使用的时候直接调用即可。

​ 在封装openpyxl时,读取excel时我写了两种实现方法,第一种是读取数据后,把每条测试数据的键值保存到一个字典中,再把字典存到一个列表,这种方式比较容易理解,但是调用的时候需要多敲几个括号或者引号;第二种是把每条测试数据的键值存到一个实例对象的属性中,再把每个实例存到一个列表。实际上他们的作用是一样,只是第二种方式调用时更简单。

read_excel.py import openpyxl class CasesData: """用于保存测试用例数据""" pass class ReadExcel: """用于读写excel数据""" def init__(self, file_name, sheet_name): """ :param file_name: excle文件名 :param sheet_name: 表单名 """ self.file_name = file_name self.sheet_name = sheet_name def open(self): """打开工作簿,选中表单""" self.wb = openpyxl.load_workbook(self.file_name) self.sh = self.wb[self.sheet_name] def close(self): """关闭工作簿对象的方法""" self.wb.close() def read_data(self): """按行读取数据,最后返回一个存储字典的列表""" self.open() # 打开工作簿 rows = list(self.sh.rows) # 按行读取所有数据 titles = [] # 用于存放用例的标题 for t in rows[0]: title = t.value titles.append(title) cases = [] # 用于存放所有的测试数据 for row in rows[1:]: case = [] # 用于存放单条测试用例数据 for r in row: case.append(r.value) cases.append(dict(zip(titles, case))) # 通过zip聚合打包用例的标题和数据 self.close() return cases def read_data_obj(self): """按行读取数据,最后返回的是一个存储实例对象的列表""" self.open() # 打开工作簿,选中表单 rows = list(self.sh.rows) # 按行读取所有数据 titles = [] # 用于存放用例的标题 for t in rows[0]: title = t.value titles.append(title) cases = [] # 用于存放所有的测试数据 for row in rows[1:]: case_obj = CasesData() # 创建一个对象用来保存该行用例数据 case = [] # 用于保存单条测试用例数据 for r in row: case.append(r.value) case_info = dict(zip(titles, case)) # 通过zip聚合打包用例的标题和数据 for k, v in case_info.items(): # 把该行测试用例标题和数据遍历出来,设成对象的属性 setattr(case_obj, k, v) cases.append(case_obj) self.close() return cases def write_data(self, row, column, msg): """写入数据的方法""" self.open() self.sh.cell(row=row, column=column, value=msg) self.wb.save(self.filename) self.close() if __name == '__main__': # 直接运行本文件时执行,下面是一个应用实例 # 需要读取excel时直接调用ReadExcel类 test = ReadExcel('cases.xlsx', 'login') res = test.read_data_obj() # 最后返回的是一个存储实例对象的列表 print(res[0].case_id) # 打印第1个用例的case_id print(res[3].title) # 打印第4个用例的title test2 = ReadExcel('cases.xlsx', 'login') res2 = test2.read_data() # 最后返回一个存储字典的列表 print(res20) # 打印第1个用例的case_id print(res23) # 打印第4个用例的title

​ 上面封装的openpyxl中,应用实例的excel表结构如下:

EOF

本文作者:miki的测试笔记
本文链接:https://www.cnblogs.com/miki-peng/p/12904383.html

相关文章
|
14天前
|
运维 数据安全/隐私保护 Python
Python基于telnetlib模块实现交换机全面巡检自动化运维项目
Python基于telnetlib模块实现交换机全面巡检自动化运维项目
46 14
|
9天前
|
监控 jenkins 持续交付
Python进行自动化
【7月更文挑战第9天】 Python在自动化部署中发挥关键作用,提供如Fabric、Ansible、Docker SDK和Kubernetes Client等工具。自动化部署提高效率、减少错误,确保部署一致性和可控性。例如,Fabric库简化了远程服务器的部署任务,如在多台服务器上执行Git拉取和Docker容器启动。持续集成/部署(CI/CD)结合Jenkins和Fabric,实现代码变更自动构建、测试和部署。监控和持续改进是确保应用稳定性和质量的关键,通过定期回顾、度量分析以及有效监控系统来优化流程。
18 1
|
10天前
|
数据格式 Python
Python代码示例,读取excel表格,将行数据转为列数据。(10)
【7月更文挑战第10天】Python代码示例,读取excel表格,将行数据转为列数据。
30 2
|
10天前
|
存储 对象存储 Python
`openpyxl`是一个用于读写Excel 2010 xlsx/xlsm/xltx/xltm文件的Python库。它不需要Microsoft Excel,也不需要.NET或COM组件。
`openpyxl`是一个用于读写Excel 2010 xlsx/xlsm/xltx/xltm文件的Python库。它不需要Microsoft Excel,也不需要.NET或COM组件。
|
12天前
|
机器学习/深度学习 存储 数据可视化
这份Excel+Python飞速搞定数据分析手册,简直可以让Excel飞起来
微软在 UserVoice 上运营着⼀个反馈论坛,每个⼈都可以在这⾥提交新点⼦供他⼈投票。票数最⾼的功能请求是“将 Python 作为Excel 的⼀门脚本语⾔”,其得票数差不多是第⼆名的两倍。尽管⾃2015 年这个点⼦发布以来并没有什么实质性进展,但在 2020 年年末,Python 之⽗ Guido van Rossum 发布推⽂称“退休太无聊了”,他将会加入微软。此事令 Excel ⽤户重燃希望。我不知道他的举动是否影响了 Excel 和 Python 的集成,但我清楚的是,为何⼈们迫切需要结合 Excel 和 Python 的⼒量,⽽你⼜应当如何从今天开始将两者结合起来。总之,这就是本
|
13天前
|
数据采集 存储 监控
python 10个自动化脚本
【7月更文挑战第10天】
33 3
|
8天前
|
监控 网络协议 安全
由于楼层自动化系统的复杂性和多样性,很难给出一个通用的Python代码示例,因为每个系统可能使用不同的硬件、通信协议和软件接口。
由于楼层自动化系统的复杂性和多样性,很难给出一个通用的Python代码示例,因为每个系统可能使用不同的硬件、通信协议和软件接口。
|
11天前
|
前端开发 JavaScript PHP
技术心得:前端点击按钮,导入excel文件,上传到后台,excel接收和更新数据
技术心得:前端点击按钮,导入excel文件,上传到后台,excel接收和更新数据
13 0
|
12天前
|
数据可视化 数据挖掘 数据处理
Python对Excel两列数据进行运算【从基础到高级的全面指南】
【7月更文挑战第6天】使用Python的`pandas`库处理Excel数据,涉及安装`pandas`和`openpyxl`,读取数据如`df = pd.read_excel('data.xlsx')`,进行运算如`df['Sum'] = df['Column1'] + df['Column2']`,并将结果写回Excel。`pandas`还支持数据筛选、分组、可视化、异常处理和性能优化。通过熟练运用这些功能,可以高效分析Excel表格。
19 0
|
14天前
|
开发者 Python
【Python】已解决:(pandas read_excel 读取Excel报错)ImportError: Pandas requires version ‘2.0.1’ or newer of ‘x
【Python】已解决:(pandas read_excel 读取Excel报错)ImportError: Pandas requires version ‘2.0.1’ or newer of ‘x
23 0