关键代码 | 相关描述 |
---|---|
cell(row, column, value) | 写入数据,三个参数分别是行,列,值 |
cell(row, column).value | 获取指定单元格的值,两个参数分别是行,列 |
workbook.remove('表单名') | 删除表单 |
del workbook['表单名'] | 删除表单 |
workbook.save('文件名') | 保存工作簿 |
sheet.max_row | 获取表单数据的总行数 |
sheet.max_column | 获取表单数据的总列数 |
sheet.rows | 获取按行所有的数据 |
sheet.columns | 获取按列所有的数据 |
🎈 注意点:
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)) # 转换成列表之后打印结果为具体的单元格,如下 # [(Cell '表单1'.A1>, Cell '表单1'.B1>, Cell '表单1'.C1>, Cell '表单1'.D1>, Cell '表单1'.E1>, Cell '表单1'.F1>, Cell '表单1'.G1>), # (Cell '表单1'.A2>, Cell '表单1'.B2>, Cell '表单1'.C2>, Cell '表单1'.D2>, Cell '表单1'.E2>, Cell '表单1'.F2>, Cell '表单1'.G2>), # (Cell '表单1'.A3>, Cell '表单1'.B3>, Cell '表单1'.C3>, Cell '表单1'.D3>, Cell '表单1'.E3>, Cell '表单1'.F3>, Cell '表单1'.G3>)] # 我们可以通过for循环以及value来查看单元格的值 for row in list(rows): # 遍历每行数据 case = [] # 用于存放一行数据 for c in row: # 把每行的每个单元格的值取出来,存放到case里 case.append(c.value) print(case)
运行结果:
C:\software\python\python.exe D:/learn/python24/python_base/test.py case_id 3 7 generator object Worksheet._cells_by_col at 0x0000012FEF90D8B8> ['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表中的原数据如下图:
在测试过程中,我们会经常需要访问excel读取测试数据,如果毎访问一次我们就调用一次openpyxl,进行打开excel,选取表单,读取数据等操作,将会增加我们的工作量和代码量。凡是需要重复使用的,我们都可以对它进行二次封装,写成我们自己的版本,还可以在封装过程中加入我们想要的内容,使用的时候直接调用即可。
在封装openpyxl时,读取excel时我写了两种实现方法,第一种是读取数据后,把每条测试数据的键值保存到一个字典中,再把字典存到一个列表,这种方式比较容易理解,但是调用的时候需要多敲几个括号或者引号;第二种是把每条测试数据的键值存到一个实例对象的属性中,再把每个实例存到一个列表。实际上他们的作用是一样,只是第二种方式调用时更简单。
# read_excel.py import openpyxl class CasesData: """用于保存测试用例数据""" pass class ReadExcel: def __init__(self, file_name, 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)) 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.file_name) 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(res2[0]['case_id']) # 打印第1个用例的case_id print(res2[3]['title']) # 打印第4个用例的title
上面封装的openpyxl中,应用实例的excel表结构如下:
以上就是python 使用openpyxl读取excel数据的详细内容,更多关于python openpyxl读取excel数据的资料请关注脚本之家其它相关文章!