在接口测试工作中,通常会将接口数据存放在excel文件中,从excel文件中读取数据,进而进行接口测试。因此,在接口测试的工作中,对excel表格的操作是十分必要的,今天就给大家介绍一下如何使用python操作excel文件。
使用python操作excel文件之前,需要电脑安装xlrd和xlwt,具体安装方法为打开doc窗口,输入指令pip install xlrd
若电脑已安装xlrd,则会提示已安装,否则显示安装过程。
同理,输入指令pip install xlwt
同理,若已安装,则提示已安装,否则显示安装过程。
安装完成后,就可以使用xlrd和xlwt操作excel文件啦!
那么如何操作呢,操作文件无非包括读文件和写文件两个操作,首先给大家介绍一下python如何读取excel文件:
1、首先定义一个类,我们所有对excel的操作都放在此类中进行:
#coding:utf-8
import xlrd
class operateExcel:
#数据初始化
def __init__(self,filePath=None,sheetId=None):
if filePath:
self.filePath=filePath
self.sheetId=sheetId
else:
self.filePath='../dataconfig/case1.xls'
self.sheetId=0
self.data=self.get_data()
#打开文件,获取到数据
def get_data(self):
data=xlrd.open_workbook(self.filePath)
table=data.sheets()[self.sheetId]
return table
#获取xls文件的行数
def getLines(self):
table=self.data
return table.nrows
#根据索引获取xls文件单个单元格的数据
def get_cell_value(self,row,col):
table=self.data
return table.cell_value(row,col)
#通过main函数进行测试
if __name__ == '__main__':
OperateExcel=operateExcel()
print(OperateExcel.getLines())
print(OperateExcel.get_cell_value(1,0))
经过测试,已正确读取excel中的数据
2、往excel中写数据
import xlwt
#创建要写的
wbk = xlwt.Workbook()
#添加sheet
sheet = wbk.add_sheet('sheet 1')
#根据索引写入内容
sheet.write(0,1,'test text')#第0行第一列写入内容
#保存文件
wbk.save('test.xls')
有人就有疑问了,我要是不想创建新的excel文件,只想读取一个现有的excel文件,并往里面添加一些数据呢,当然没问题了:
def write_value(self,row,col,value):
#根据row,col,value写入数据
read_data = xlrd.open_workbook(self.file_name)
write_data = copy(read_data)
sheet_data = write_data.get_sheet(0)
sheet_data.write(row,col,value)
write_data.save(self.file_name)
通过索引来取值确实很方便,但是有时候excel表格中那么多数据,我总不能都记得他的索引是什么吧?今天就先到这里,明天有时间再给大家介绍一下如何更简洁高效的读取excel文件。