一、背景
python自动化测试中,有时候也需要使用到Excel文件,而此时要借用第三方库才能进行操作。在执行读execl文件时要用到xlrd库,要写入数据至excel文件时,要用到xlwt库,它们需要单独安装,安装方式也非常简单。当然也可以用其他的第三方库,比如openpyxl和xlutils库。最后,选择什么模块要根据实际情况而定。
二、各模块之间的区别
xlrd,xlwt和xlutils是用Python处理Excel文档(*.xls)的高效率工具。其中,xlrd只能读取xls,xlwt只能新建xls(不可以修改),xlutils能将xlrd.Book转为xlwt.Workbook,从而得以在现有xls的基础上修改数据,并创建一个新的xls,实现修改。openpyxl(可读写excel表)专门处理Excel2007及以上版本产生的xlsx文件,xls和xlsx之间转换容易。另外,还有以下特点。
- 读取文件速度,xlrd比openpyxl较快
- xlwt写入文件效率较高
- xlwt写入数据会有局限性,不能写入超过65535行、256列的数据(因为它只支持Excel 2003及之前的版本,在这些版本的Excel中行数和列数有此限制)
- xlrd,xlwt和xlutils是用Python处理Excel文档(*.xls)的高效率工具。其中,xlrd只能读取xls,xlwt只能新建xls(不可以修改),xlutils能将xlrd.Book转为xlwt.Workbook,从而得以在现有xls的基础上修改数据
- xlwt不支持07版以后的excel并创建一个新的xlsx(可以创建xls格式的excel文件),openpyxl支持07+的excel,一直有人在维护,文档清晰易读。
三、安装第三方库
这里选择xlrd,xlwt模块进行简单介绍。
安装xlrd库:
pip install xlrd
安装xlwt库:
pip install xlwt
也可以直接在pycharm编辑器中安装,方法也非常的简单:先打开settings,然后找到project interpreter,最后再添加想要安装的模块即可。
四、读取excel文件数据
这里准备了一个招聘人员信息表,名字叫zhaopin.xlsx的工作簿,其中有两个表,名字叫招聘人员和数据分析。如图:
下面我们用python第三方库xlrd来读取工作簿中的信息。
1、 sheet_names
:获取所有的sheet
的名字。
import xlrd
workbook = xlrd.open_workbook('zhaopin.xlsx') # 打开名字叫'zhaopin.xlsx'的工作簿
sheet_names = workbook.sheet_names() # 获得工作簿中的所有表明
print(sheet_names)
结果:
['招聘人员', '数据分析']
2、 sheet_by_index
:根据索引获取sheet
对象。
import xlrd
workbook = xlrd.open_workbook('zhaopin.xlsx') # 打开名字叫'zhaopin.xlsx'的工作簿
sheet0 = workbook.sheet_by_index(0) # 获取索引为0的表对象,0对应的是excel工作簿中的第一张表
sheet1 = workbook.sheet_by_index(1) # 获取索引为1的表对象,1对应的是excel工作簿中的第二张表
print(sheet0.name)
print(sheet1.name)
结果:
招聘人员
数据分析
3、 sheet_by_name
:根据名字获取sheet
对象。
import xlrd
workbook = xlrd.open_workbook('zhaopin.xlsx') # 打开名字叫'zhaopin.xlsx'的工作簿
sheet = workbook.sheet_by_name('数据分析')
print(sheet.name)
结果:
数据分析
注意:sheet_by_name('数据分析')括号中的表名称一定要填写正确,否则无法找到对象,会报错。
4、sheets
:获取所有的sheet
对象。
import xlrd
workbook = xlrd.open_workbook('zhaopin.xlsx') # 打开名字叫'zhaopin.xlsx'的工作簿
sheets = workbook.sheets() # 获取所有的`sheet`对象
for sheet in sheets:
print(sheet.name)
结果:
招聘人员
数据分析
5、sheet.nrows
:这个sheet
中的行数。
6、sheet.ncols
:这个sheet
中的列数。
import xlrd
workbook = xlrd.open_workbook('zhaopin.xlsx') # 打开名字叫'zhaopin.xlsx'的工作簿
sheet = workbook.sheet_by_index(0) # 获取索引为0的表对象,0对应的是excel工作簿中的第一张表
sheet_nrows = sheet.nrows # 这个`sheet`中的行数
sheet_ncols = sheet.ncols # 这个`sheet`中的列数
print('行数为:{}'.format(sheet_nrows))
print('列数为:{}'.format(sheet_ncols))
结果:
行数为:78
列数为:7
7、sheet.cell(row,col):获取指定行和列的cell对象。
import xlrd
workbook = xlrd.open_workbook('zhaopin.xlsx') # 打开名字叫'zhaopin.xlsx'的工作簿
sheet = workbook.sheet_by_name('招聘人员')
cells = sheet.cell(0,1) # 获取指定0行和1列的cell对象。
print(cells.value) # 打印这个cell里面的值。
结果:
姓名
8、sheet.row_slice(row,start_col,end_col):获取指定行的某几列的cell对象。
import xlrd
workbook = xlrd.open_workbook('zhaopin.xlsx') # 打开名字叫'zhaopin.xlsx'的工作簿
sheet = workbook.sheet_by_name('招聘人员')
cells = sheet.row_slice(0,0,6) # 获取指定0行的0列到5列的cell对象。
print(cells)
结果:
[text:'序号', text:'姓名', text:'性别', text:'专业', text:'学历', text:'毕业院校']
9、sheet.col_slice(col,start_row,end_row):获取指定列的某几行的cell对象。
import xlrd
workbook = xlrd.open_workbook('zhaopin.xlsx') # 打开名字叫'zhaopin.xlsx'的工作簿
sheet = workbook.sheet_by_index(0) # 获取索引为0的表对象,0对应的是excel工作簿中的第一张表
cells = sheet.col_slice(1,0,5)#获取1列0行到4行的cell对象,返回一个列表
for cell in cells:
print(cell.value) # 打印这个cell里面的值。
结果:
姓名
冯涛
李佳楠
罗旭
张自航
10、sheet.cell_value(row,col):获取指定行和列的值。
import xlrd
workbook = xlrd.open_workbook('zhaopin.xlsx') # 打开名字叫'zhaopin.xlsx'的工作簿
sheet = workbook.sheet_by_index(0)
cells = sheet.cell_value(0,5)
print(cells)
结果:
毕业院校
11、sheet.row_values(row,start_col,end_col):获取指定行的某几列的值。
import xlrd
workbook = xlrd.open_workbook('zhaopin.xlsx') # 打开名字叫'zhaopin.xlsx'的工作簿
sheet = workbook.sheet_by_index(0)
cells = sheet.row_values(0,0,6) # 获取0行从0列到5列的值。
print(cells)
结果:
['序号', '姓名', '性别', '专业', '学历', '毕业院校']
注意:它与sheet.row_slice(row,start_col,end_col)的区别,一个是获取对象,一个是获取值。
12、sheet.col_values(col,start_row,end_row):获取指定列的某几行的值。
import xlrd
workbook = xlrd.open_workbook('zhaopin.xlsx') # 打开名字叫'zhaopin.xlsx'的工作簿
sheet = workbook.sheet_by_index(0)
cells = sheet.col_values(1,0,5)
print(cells)
结果:
['姓名', '冯涛', '李佳楠', '罗旭', '张自航']
注意:它与sheet.col_slice(col,start_row,end_row)的区别,一个是获取对象,一个是获取值。
五、写入数据至excel文件
1、cell的数据类型
-
xlrd.XL_CELL_TEXT(Text)
:文本类型。 -
xlrd.XL_CELL_NUMBER(Number)
:数值类型。 -
xlrd.XL_CELL_DATE(Date)
:日期时间类型。 -
xlrd.XL_CELL_BOOLEAN(Bool)
:布尔类型。 -
xlrd.XL_CELL_BLANK
:空白数据类型。
2、写入数据至excel文件步骤
- 首先,导入
xlwt
模块。 - 然后,创建一个
Workbook
对象。 - 然后,创建一个
Sheet
对象。 - 接着,使用
sheet.write(row,col,data)
方法把数据写入到Sheet
下指定行和列中。如果想要在原来workbook
对象上添加新的cell
,那么需要调用put_cell
来添加。 - 最后,保存成
Excel
文件。
3、实例
比如,我想在原来的‘zhaopin.xlsx’表中添加"总分"一列,并在其中随机填充面试得分。需要注意,如果想要在原来已经存在的Excel
文件中添加新的行或者新的列,那么需要采put_cell(row,col,type,value,xf_index)
来添加进去,最后再放到xlwt
创建的workbook
中,然后再保存进去。示例代码如下:
import xlwt
import xlrd
import random
workbook = xlrd.open_workbook('zhaopin.xlsx') # 打开名字叫'zhaopin.xlsx'的工作簿
rsheet = workbook.sheet_by_index(0)
rsheet.put_cell(0, 6, xlrd.XL_CELL_TEXT, '总分', None) # 在0行6列处添加“总分”列,列属性为文本
for row in range(1, rsheet.nrows):
rsheet.put_cell(
row,
6,
xlrd.XL_CELL_TEXT,
random.randint(
70,
99),
None) # 随机向每行6列添加70~99之间的数
write_workbook = xlwt.Workbook(encoding='utf8') # 重新创建一个工作簿
wsheet = write_workbook.add_sheet('招聘人员') # 工作簿中创建一个名字为“招聘人员”的表
for row in range(rsheet.nrows):
for col in range(rsheet.ncols):
wsheet.write(row, col, rsheet.cell_value(row, col)) # 将原表中的数据写入新创建的表中
write_workbook.save('zhaopin_new.xls') # 保存
结果: