赞
踩
对比其它编程语言,我们都知道Python最大的优势是代码简单,有丰富的第三方开源库供开发者使用。伴随着近几年数据分析的热度,Python也成为最受欢迎的编程语言之一。而对于数据的读取和存储,对于普通人来讲,除了数据库之外,最常见的就是微软的Excel。
Microsoft Excel是Microsoft为使用Windows和Apple Macintosh操作系统的电脑编写的一款电子表格软件。
Excel中有xls和xlsx两种格式,它们之间的区别是:
Python自带的模块中有针对xls格式的xlrd和xlwt模块,但这两个库仅仅是针对xls的操作,当我们要操作xlsx格式文件时,则需要使用到openpyxl第三方库。
当使用以上几个模块的时候,从理论上我们就可以完全操作不同格式的Excel的读和写,很多人就疑惑,那这篇文章的作用是什么?我们直接学习对应的这三个模块不就好了吗?
答案就是:虽然这几个库已经把Excel的文件、表、行、列的概念完全转换为Python中的对象,但每次操作都需要遍历每一个单元格,甚至很多时候我们要花费大量的时间在思考循环单元格的边界上,这本身就是在重复造轮子,因此我花了半天时间整理了以下六个函数。
def read_xls_excel(url,index): ''' 读取xls格式文件 参数: url:文件路径 index:工作表序号(第几个工作表,传入参数从1开始数) 返回: data:表格中的数据 ''' # 打开指定的工作簿 workbook = xlrd.open_workbook(url) # 获取工作簿中的所有表格 sheets = workbook.sheet_names() # 获取工作簿中所有表格中的的第 index 个表格 worksheet = workbook.sheet_by_name(sheets[index-1]) # 定义列表存储表格数据 data = [] # 遍历每一行数据 for i in range(0, worksheet.nrows): # 定义表格存储每一行数据 da = [] # 遍历每一列数据 for j in range(0, worksheet.ncols): # 将行数据存储到da列表 da.append(worksheet.cell_value(i, j)) # 存储每一行数据 data.append(da) # 返回数据 return data
def write_xls_excel(url,sheet_name,two_dimensional_data): ''' 写入xls格式文件 参数: url:文件路径 sheet_name:表名 two_dimensional_data:将要写入表格的数据(二维列表) ''' # 创建工作簿对象 workbook = xlwt.Workbook() # 创建工作表对象 sheet = workbook.add_sheet(sheet_name) # 遍历每一行数据 for i in range(0,len(two_dimensional_data)): # 遍历每一列数据 for j in range(0,len(two_dimensional_data[i])): # 写入数据 sheet.write(i,j,two_dimensional_data[i][j]) # 保存 workbook.save(url) print("写入成功")
def write_xls_excel_add(url, two_dimensional_data, index): ''' 追加写入xls格式文件 参数: url:文件路径 two_dimensional_data:将要写入表格的数据(二维列表) index:指定要追加的表的序号(第几个工作表,传入参数从1开始数) ''' # 打开指定的工作簿 workbook = xlrd.open_workbook(url) # 获取工作簿中的所有表格 sheets = workbook.sheet_names() # 获取指定的表 worksheet = workbook.sheet_by_name(sheets[index-1]) # 获取表格中已存在的数据的行数 rows_old = worksheet.nrows # 将xlrd对象拷贝转化为xlwt对象 new_workbook = copy(workbook) # 获取转化后工作簿中的第index个表格 new_worksheet = new_workbook.get_sheet(index-1) # 遍历每一行数据 for i in range(0, len(two_dimensional_data)): # 遍历每一列数据 for j in range(0, len(two_dimensional_data[i])): # 追加写入数据,注意是从i+rows_old行开始写入 new_worksheet.write(i+rows_old, j, two_dimensional_data[i][j]) # 保存工作簿 new_workbook.save(url) print("追加写入成功")
def read_xlsx_excel(url, sheet_name): ''' 读取xlsx格式文件 参数: url:文件路径 sheet_name:表名 返回: data:表格中的数据 ''' # 使用openpyxl加载指定路径的Excel文件并得到对应的workbook对象 workbook = openpyxl.load_workbook(url) # 根据指定表名获取表格并得到对应的sheet对象 sheet = workbook[sheet_name] # 定义列表存储表格数据 data = [] # 遍历表格的每一行 for row in sheet.rows: # 定义表格存储每一行数据 da = [] # 从每一行中遍历每一个单元格 for cell in row: # 将行数据存储到da列表 da.append(cell.value) # 存储每一行数据 data.append(da) # 返回数据 return data
def write_xlsx_excel(url, sheet_name, two_dimensional_data): ''' 写入xlsx格式文件 参数: url:文件路径 sheet_name:表名 two_dimensional_data:将要写入表格的数据(二维列表) ''' # 创建工作簿对象 workbook = openpyxl.Workbook() # 创建工作表对象 sheet = workbook.active # 设置该工作表的名字 sheet.title = sheet_name # 遍历表格的每一行 for i in range(0, len(two_dimensional_data)): # 遍历表格的每一列 for j in range(0, len(two_dimensional_data[i])): # 写入数据(注意openpyxl的行和列是从1开始的,和我们平时的认知是一样的) sheet.cell(row=i + 1, column=j + 1, value=str(two_dimensional_data[i][j])) # 保存到指定位置 workbook.save(url) print("写入成功")
def write_xlsx_excel_add(url, sheet_name, two_dimensional_data): ''' 追加写入xlsx格式文件 参数: url:文件路径 sheet_name:表名 two_dimensional_data:将要写入表格的数据(二维列表) ''' # 使用openpyxl加载指定路径的Excel文件并得到对应的workbook对象 workbook = openpyxl.load_workbook(url) # 根据指定表名获取表格并得到对应的sheet对象 sheet = workbook[sheet_name] for tdd in two_dimensional_data: sheet.append(tdd) # 保存到指定位置 workbook.save(url) print("追加写入成功")
希望这篇文章对正在学习Python的你有一定的帮助!
感兴趣的小伙伴,赠送全套Python学习资料,包含面试题、简历资料等具体看下方。
声明:本文内容由网友自发贡献,转载请注明出处:【wpsshop博客】
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。