python读写excel文件

一、背景

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')  # 保存

结果:

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 194,242评论 5 459
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 81,769评论 2 371
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 141,484评论 0 319
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 52,133评论 1 263
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 61,007评论 4 355
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 46,080评论 1 272
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 36,496评论 3 381
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 35,190评论 0 253
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 39,464评论 1 290
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 34,549评论 2 309
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 36,330评论 1 326
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 32,205评论 3 312
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 37,567评论 3 298
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 28,889评论 0 17
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 30,160评论 1 250
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 41,475评论 2 341
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 40,650评论 2 335

推荐阅读更多精彩内容