前言
使用 pandas 在 django 框架中生成 excel 文件返回
一、pandas是什么?
pandas 是基于NumPy 的一种工具,该工具是为解决数据分析任务而创建的。Pandas 纳入了大量库和一些标准的数据模型,提供了高效地操作大型数据集所需的工具。pandas提供了大量能使我们快速便捷地处理数据的函数和方法。你很快就会发现,它是使Python成为强大而高效的数据分析环境的重要因素之一
二、使用步骤
1.引入库
代码如下(示例):
import pandas as pd
from io import BytesIO
from xlwt import Workbook #用户生成工作簿
from datetime import datetime
from django.utils.http import urlquote
from django.http import HttpResponse
2.具体代码
代码如下(示例):
def create_excel(data, name):
"""
data 导出的数据
name excel 名称
"""
wb = Workbook() # 生成一个工作簿(即一个Excel文件)
wb.encoding = 'utf-8'
base_pd = pd.DataFrame(data)
base_pd.rename(columns=self.get_excel_title(), inplace=True) # 对列表名字进行重新定义
# 准备写入到IO中
output = BytesIO()
writer = pd.ExcelWriter(output)
base_pd.to_excel(writer, name, encoding='utf-8', index=None)
output.seek(0) # 重新定位到开始
writer.save() # 将Excel文件内容保存到IO中
# 设置HttpResponse的类型
response = HttpResponse(content_type='text/csv')
response.write(codecs.BOM_UTF8)
file_name = '{}_{}.xls'.format(name, datetime.now().strftime('%Y-%m-%d %H:%M:%S')) # 给文件名中添加日期时间
file_name = urlquote(file_name) # 使用urlquote()方法解决中文无法使用的问题
response.content = output.getvalue()
response['Content-Disposition'] = 'attachment; filename=%s' % file_name
return response
总结
使用 xlwt 或者 openpyxl 在写入 excel 时候循环使用,虽然使用 pandas 导出 excel 大材小用,但对于过多的数据还是挺快的,也节省了开发时间。