Django实现按列写入csv文件并提供下载

短命女 2024-04-08 09:58 72阅读 0赞

**zip\_longest(\*)**将所有输入的对象行转置为列，返回长度与最长的对象相同的列表，运用此方法，writer对象会将原列表以列的形式写入csv文件。

def test(request):
        from itertools import zip_longest
        list1=[1,2]
        list2=[3,4,5]
        list_head=['list1','list2']
        list_value=[list1,list2]
        export_data = zip_longest(*list_value, fillvalue='')
        # 会在writer对象中将列表以列的形式写入csv文件
    
        response = HttpResponse(content_type='text/csv')#定义一个HttpResponse，类型是csv
        response['Content-Disposition'] = "attachment;filename=file.csv"#定义返回的信息，以附件方式下载及文件名称；
        import csv
        wr = csv.writer(response)
        wr.writerow(list_head)
        wr.writerows(export_data)
    
        return response

对于生成数据量不大的文件，可以使用HttpResponse将数据存入内存然后再输出：

1.  response = HttpResponse(content\_type='text/csv') 定义一个HttpResponse，类型为csv格式的文件；如果用默认值（html），那么最终会返回html格式的文件。
2.  response\['Content-Disposition'\] = "attachment;filename=file.csv" attachment表示将此文件作为附件的形式下载，filename=file.csv用来指定csv文件的名字。（**注意file.csv不要加引号，有可能输出格式错误**）

测试：

输入网址：![487c9113414142f5aff6f40c150ae507.png][]

打开文件：

![cef9e8dfbaf341fea4399e873058168e.png][]

参考链接：[将Python列表写入csv中的列 - 问答 - 腾讯云开发者社区-腾讯云 (tencent.com)][Python_csv_ - _ - _-_ _tencent.com]

[487c9113414142f5aff6f40c150ae507.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/04/08/f120cdf1c31945e1bc020662750ef4e3.png
[cef9e8dfbaf341fea4399e873058168e.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/04/08/258e721c6f7443359e8d2f56596feb59.png
[Python_csv_ - _ - _-_ _tencent.com]: https://cloud.tencent.com/developer/ask/sof/32347