使用python将大量数据导出到Excel中的小技巧分享
- 1、下载文档前请自行甄别文档内容的完整性,平台不提供额外的编辑、内容补充、找答案等附加服务。
- 2、"仅部分预览"的文档,不可在线预览部分如存在完整性等问题,可反馈申请退款(可完整预览的文档不适用该条件!)。
- 3、如文档侵犯您的权益,请联系客服反馈,我们会尽快为您处理(人工客服工作时间:9:00-18:30)。
使⽤python将⼤量数据导出到Excel中的⼩技巧分享
(1)问题描述:为了更好地展⽰数据,Excel格式的数据⽂件往往⽐⽂本⽂件更具有优势,但是具体到python中,该如何导出数据到Excel呢?如果碰到需要导出⼤量数据⼜该如何操作呢?
本⽂主要解决以上两个问题。
(2)具体步骤如下:
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
2.第⼆步,哈哈,没有啦,废话不说了,直接上代码,ps,代码中包含xlwt和openpyxl的两个实现版本。
(3)扩展阅读:通过查阅资料,发现⽹上众说纷纭,总结起来有如下⼏点:
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的情况,你将有如下三种选择,
(1) 换⼀种存储格式,如保存为CSV⽂件
(2) 使⽤openpyxl—,因为它⽀持对Excel 2007+ xlsx/xlsm format的处理
(3) win32 COM (Windows only)
当然,我们要直⾯困难了,为了更好地展⽰数据给产品和⽤户,我们依然选择的第⼆种。
ps,⾮常lucky,⼀番搜索后我找到了,⽀持07+的excel,⼀直有⼈在维护,⽂档清晰易读,参照Tutorial和API⽂档很快就能上⼿,就是它了~
(4) 闲话少说,直接上代码,敬请参考
# coding:utf-8
'''
# 希望对⼤家有帮助哈,请多提问题
create by yaoyz
date: 2017/01/24
'''
import xlrd
import xlwt
# workbook相关
from openpyxl.workbook import Workbook
# ExcelWriter,封装了很强⼤的excel写的功能
from openpyxl.writer.excel import ExcelWriter
# ⼀个eggache的数字转为列字母的⽅法
from openpyxl.utils import get_column_letter
from openpyxl.reader.excel import load_workbook
class HandleExcel():
'''Excel相关操作类'''
def __init__(self):
self. head_row_labels = [u'学⽣ID',u'学⽣姓名',u'联系⽅式',u'知识点ID',u'知识点名称']
"""
function:
读出txt⽂件中的每⼀条记录,把它保存在list中
Param:
filename: 要读出的⽂件名
Return:
res_list:返回的记录的list
"""
def read_from_file(self,filename):
res_list=[]
file_obj=open(filename,"r")
for line in file_obj.readlines():
res_list.append(line)
file_obj.close()
return res_list
"""
function:
读出*.xlsx中的每⼀条记录,把它保存在data_dic中返回
Param:
excel_name: 要读出的⽂件名
Return:
data_dic:返回的记录的dict
"""
def read_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.title
print "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
# 把数据存到字典中
for row in ws.rows:
temp_list=[]
# print "row",row
for cell in row:
print cell.value,
temp_list.append(cell.value)
print ""
data_dic[sign]=temp_list
sign+=1
print data_dic
return data_dic
"""
function:
读出*.xlsx中的每⼀条记录,把它保存在data_dic中返回
Param:
records: 要保存的,⼀个包含每⼀条记录的list
save_excel_name: 保存为的⽂件名
head_row_stu_arrive_star:
Return:
data_dic:返回的记录的dict
"""
def write_to_excel_with_openpyxl(self,records,head_row,save_excel_name="save.xlsx"): # 新建⼀个workbook
wb = Workbook()
# 新建⼀个excelWriter
ew = ExcelWriter(workbook=wb)
# 设置⽂件输出路径与名称
dest_filename = save_excel_name.decode('utf-8')
# 第⼀个sheet是ws
ws = wb.worksheets[0]
# 设置ws的名称
ws.title = "range names"
# 写第⼀⾏,标题⾏
for h_x in range(1,len(head_row)+1):
h_col=get_column_letter(h_x)
#print h_col
ws.cell('%s%s' % (h_col, 1)).value = '%s' % (head_row[h_x-1])
# 写第⼆⾏及其以后的那些⾏
i = 2
for record in records:
record_list=str(record).strip().split("\t")
for x in range(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:
⽆
"""
def read_excel(self,excel_name):
workbook=xlrd.open_workbook(excel_name)
print workbook.sheet_names()
# 获取所有sheet
print workbook.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的名称,⾏数,列数
print ,sheet2.nrows,sheet2.ncols
# 获取整⾏和整列的值(数组)
rows = sheet2.row_values(3) # 获取第四⾏内容
cols = sheet2.col_values(2) # 获取第三列内容
print rows
print cols
# 获取单元格内容
print sheet2.cell(1,0).value
print sheet2.cell_value(1,0)
print sheet2.row(1)[0].value
# 获取单元格内容的数据类型
print sheet2.cell(1,0).ctype
# 通过名称获取
return workbook.sheet_by_name(u'Sheet1')
"""
function:
设置单元格样式
Param:
name: 字体名字
height: 字体⾼度
bold: 是否⼤写
Return:
style: 返回设置好的格式对象
"""
def set_style(self,name,height,bold=False):
style = xlwt.XFStyle() # 初始化样式
font = xlwt.Font() # 为样式创建字体
= name # 'Times New Roman'
font.bold = bold
font.color_index = 4
font.height = height
borders= xlwt.Borders()
borders.left= 6
borders.right= 6
borders.top= 6
borders.bottom= 6
style.font = font
style.borders = borders
return style
"""
function:
按照设置单元格样式把计算结果由txt转变为Excel存储
Param:
dataset:要保存的结果数据,list存储
Return:
将结果保存为 excel对象中
"""
def write_to_excel(self, dataset,save_excel_name,head_row):
f = xlwt.Workbook() # 创建⼯作簿
# 创建第⼀个sheet:
# sheet1
count=1
sheet1 = f.add_sheet(u'sheet1', cell_overwrite_ok=True) # 创建sheet
# ⾸⾏标题:
for p in range(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 work
for line in dataset:
row_list=str(line).strip("\n").split("\t")
for pp in range(len(str(line).strip("\n").split("\t"))):
sheet1.write(count,pp,row_list[pp].decode('utf-8'),default)
count+=1
f.save(save_excel_name) # 保存⽂件
def run_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_labels
save_name="test_openpyxl.xlsx"
self.write_to_excel_with_openpyxl(dataset_list,head_row_label,save_name)
print "3. 执⾏完毕,由txt格式⽂件保存为Excel⽂件的任务"
def run_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_labels
save_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中的⼩技巧分享就是⼩编分享给⼤家的全部内容了,希望能给⼤家⼀个参考,也希望⼤家多多⽀持。