社区所有版块导航
Python
python开源   Django   Python   DjangoApp   pycharm  
DATA
docker   Elasticsearch  
aigc
aigc   chatgpt  
WEB开发
linux   MongoDB   Redis   DATABASE   NGINX   其他Web框架   web工具   zookeeper   tornado   NoSql   Bootstrap   js   peewee   Git   bottle   IE   MQ   Jquery  
机器学习
机器学习算法  
Python88.com
反馈   公告   社区推广  
产品
短视频  
印度
印度  
Py学习  »  Python

【好消息】Python和Excel终于可以互通了!!

数据分析1480 • 5 年前 • 588 次点击  
今天小编为大家分享一篇使用python将大量数据导出到Excel中的技巧心得,可以让Python和Excel的数据实现互通!具有很好的参考价值,希望对大家有所帮助(建议在电脑端阅读,代码案例较多)。一起跟随小编过来看看吧!



问题描述


为了更好地展示数据,Excel格式的数据文件往往比文本文件更具有优势,但是具体python中,该如何导出数据到Excel呢?如果碰到需要导出大量数据又该如何操作呢?



具体步骤


Step 1 安装openpyxl


使用pip install openpyxl即可,但是在windows下安装的是2.2.6版本,在centos自动安装的是4.1版本,写的代码在windows下运行没问题,但centos上却报错了,说是ew=ExcelWriter(workbook=wb)少提供一个参数,于是果断在 237服务器上我已安装2.2.6版本的,问题解决。


pip install openpyxl==2.2.6


Step 2 直接上代码(Ps:代码中包含xlwt和openpyxl的两个实现版本)


# coding:utf-8

'''# 希望对大家有帮助哈,请多提问题create by yaoyzdate: 2017/01/24'''importxlrdimportxlwt# workbook相关fromopenpyxl.workbookimportWorkbook# ExcelWriter,封装了很强大的excel写的功能fromopenpyxl.writer.excelimportExcelWriter# 一个eggache的数字转为列字母的方法fromopenpyxl.utilsimportget_column_letterfromopenpyxl.reader.excelimportload_workbookclassHandleExcel():'''Excel相关操作类'''def__init__(self):self. head_row_labels = [u'学生ID',u'学生姓名',u'联系方式',u'知识点ID',u'知识点名称']"""function:读出txt文件中的每一条记录,把它保存在list中Param:filename: 要读出的文件名Return:res_list:返回的记录的list"""defread_from_file(self,filename):res_list=[]file_obj=open(filename,"r")forlineinfile_obj.readlines():res_list.append(line)file_obj.close()returnres_list""" function:读出*.xlsx中的每一条记录,把它保存在data_dic中返回Param:excel_name: 要读出的文件名Return:data_dic:返回的记录的dict"""defread_excel_with_openpyxl(self, excel_name="testexcel2007.xlsx"):# 读取excel2007文件wb = load_workbook(filename=excel_name)# 显示有多少张表print"Worksheet range(s):", wb.get_named_ranges()print"Worksheet name(s):", wb.get_sheet_names()# 取第一张表sheetnames = wb.get_sheet_names()ws = wb.get_sheet_by_name(sheetnames[0])# 显示表名,表行数,表列数print"Work Sheet Titile:",ws.titleprint"Work Sheet Rows:",ws.get_highest_row()print"Work Sheet Cols:",ws.get_highest_column()# 获取读入的excel表格的有多少行,有多少列row_num=ws.get_highest_row()col_num=ws.get_highest_column()print"row_num: ",row_num," col_num: ",col_num# 建立存储数据的字典data_dic = {}sign=1# 把数据存到字典中forrowinws.rows:temp_list=[]# print "row",rowforcellinrow:printcell.value,temp_list.append(cell.value)print""data_dic[sign]=temp_listsign+=1printdata_dicreturndata_dic"""function:读出*.xlsx中的每一条记录,把它保存在data_dic中返回Param:records: 要保存的,一个包含每一条记录的listsave_excel_name: 保存为的文件名head_row_stu_arrive_star:Return:data_dic:返回的记录的dict"""defwrite_to_excel_with_openpyxl(self,records,head_row,save_excel_name="save.xlsx"):# 新建一个workbookwb = Workbook()# 新建一个excelWriterew = ExcelWriter(workbook=wb)# 设置文件输出路径与名称dest_filename = save_excel_name.decode('utf-8')# 第一个sheet是wsws = wb.worksheets[0]# 设置ws的名称ws.title ="range names"# 写第一行,标题行forh_xinrange(1,len(head_row)+1):h_col=get_column_letter(h_x)#print h_colws.cell('%s%s'% (h_col,1)).value ='%s'% (head_row[h_x-1])# 写第二行及其以后的那些行i =2forrecordinrecords:record_list=str(record).strip().split("\t")forxinrange(1,len(record_list)+1):col = get_column_letter(x)ws.cell('%s%s'% (col, i)).value ='%s'% (record_list[x-1].decode('utf-8'))i +=1# 写文件ew.save(filename=dest_filename)"""function:测试输出Excel内容读出Excel文件Param:excel_name: 要读出的Excel文件名Return:"""defread_excel(self,excel_name):workbook=xlrd.open_workbook(excel_name)printworkbook.sheet_names()# 获取所有sheetprintworkbook.sheet_names()# [u'sheet1', u'sheet2']sheet2_name = workbook.sheet_names()[1]# 根据sheet索引或者名称获取sheet内容sheet2 = workbook.sheet_by_index(1)# sheet索引从0开始sheet2 = workbook.sheet_by_name('Sheet1')# sheet的名称,行数,列数printsheet2.name,sheet2.nrows,sheet2.ncols # 获取整行和整列的值(数组)rows = sheet2.row_values(3)# 获取第四行内容cols = sheet2.col_values(2)# 获取第三列内容printrowsprintcols# 获取单元格内容printsheet2.cell(1,0).valueprintsheet2.cell_value(1,0)printsheet2.row(1)[0].value# 获取单元格内容的数据类型printsheet2.cell(1,0).ctype# 通过名称获取returnworkbook.sheet_by_name(u'Sheet1')"""function:设置单元格样式Param:name:  字体名字height: 字体高度bold: 是否大写Return:style: 返回设置好的格式对象"""defset_style(self,name,height,bold=False):style = xlwt.XFStyle()# 初始化样式font = xlwt.Font()# 为样式创建字体font.name = name# 'Times New Roman'font.bold = boldfont.color_index =4font.height = heightborders= xlwt.Borders()borders.left=6borders.right=6borders.top=6borders.bottom=6style.font = fontstyle.borders = bordersreturnstyle"""function:按照 设置单元格样式 把计算结果由txt转变为Excel存储Param:dataset:要保存的结果数据,list存储Return:将结果保存为 excel对象中"""defwrite_to_excel(self, dataset,save_excel_name,head_row):f = xlwt.Workbook()# 创建工作簿# 创建第一个sheet:# sheet1count=1sheet1 = f.add_sheet(u'sheet1', cell_overwrite_ok=True)# 创建sheet# 首行标题:forpinrange(len(head_row)):sheet1.write(0,p,head_row[p],self.set_style('Times New Roman',250,True))default=self.set_style('Times New Roman',200,False)# define style out the loop will workforlineindataset:row_list=str(line).strip("\n").split("\t")forppinrange(len(str(line).strip("\n").split("\t"))):sheet1.write(count,pp,row_list[pp].decode('utf-8'),default)count+=1f.save(save_excel_name)# 保存文件defrun_main_save_to_excel_with_openpyxl(self):print"测试读写2007及以后的excel文件xlsx,以方便写入文件更多数据"print"1. 把txt文件读入到内存中,以list对象存储"dataset_list=self.read_from_file("test_excel.txt")'''test use openpyxl to handle EXCEL 2007'''print"2. 把文件写入到Excel表格中"head_row_label=self.head_row_labelssave_name="test_openpyxl.xlsx"self.write_to_excel_with_openpyxl(dataset_list,head_row_label,save_name)print"3. 执行完毕,由txt格式文件保存为Excel文件的任务"defrun_main_save_to_excel_with_xlwt(self):print" 4. 把txt文件读入到内存中,以list对象存储"dataset_list=self.read_from_file("test_excel.txt")'''test use xlwt to handle EXCEL 97-2003'''print" 5. 把文件写入到Excel表格中"head_row_label=self.head_row_labelssave_name="test_xlwt.xls"self.write_to_excel_with_openpyxl(dataset_list,head_row_label,save_name)print"6. 执行完毕,由txt格式文件保存为Excel文件的任务"if__name__ =='__main__':print"create handle Excel Object"obj_handle_excel=HandleExcel()# 分别使用openpyxl和xlwt将数据写入文件obj_handle_excel.run_main_save_to_excel_with_openpyxl() obj_handle_excel.run_main_save_to_excel_with_xlwt()'''测试读出文件,注意openpyxl不可以读取xls的文件,xlrd不可以读取xlsx格式的文件'''#obj_handle_excel.read_excel_with_openpyxl("testexcel2003.xls") # 错误写法#obj_handle_excel.read_excel_with_openpyxl("testexcel2003.xls") # 错误写法obj_handle_excel.read_excel("testexcel2003.xls")obj_handle_excel.read_excel_with_openpyxl("testexcel2007.xlsx")



 扩展阅读


通过查阅资料,发现网上众说纷纭,总结起来有如下几点:


python Excel相关操作的module lib有两组,一组是xlrd、xlwt、xlutils,另一组是openpyxl,但是前一组(xlrd,xlwt)比较老,只能处理由Excel 97-2003 或者Excel 97 以前版本生成的xls格式的excel文件,xlwt甚至不支持07版以后的excel,这个格式excel文件一般来说,最大只能支持256列或者65536行的excel文件。


因此面对需要导出大量数据到excel的情况,你将有如下三种选择:


  • 换一种存储格式,如保存为CSV文件

  • 使用openpyxl—,因为它支持对Excel 2007+ xlsx/xlsm format的处理

  • win32 COM (Windows only)


当然,我们要直面困难,为了更好地展示数据给产品和用户,我们依然选择第二种。


经过一番搜索后我找到了openpyxl的网址,放在下面了,支持07+的excel,一直有人在维护,文档清晰易读,参照Tutorial和API文档很快就能上手了,大家有需要的可以自取。


  • openpyxl网址:

https://openpyxl.readthedocs.io/en/stable/(可以将网页翻译成中文方便使用)


End.

作者:小天真_5eed

来源:简书

本文为转载分享,如有侵权请联系后台删除

Python社区是高质量的Python/Django开发社区
本文地址:http://www.python88.com/topic/62260
 
588 次点击