如何直接将csv文件压缩成zip压缩包?

v1l68za4  于 2022-12-06  发布在  其他
关注(0)|答案(3)|浏览(181)

我正在使用以下代码动态生成许多csv文件:

import csv
fieldnames = ['foo1', 'foo2', 'foo3', 'foo4']
with open(csvfilepath, 'wb') as csvfile:
    csvwrite = csv.DictWriter(csvfile, delimiter=',', fieldnames=fieldnames)
    csvwrite.writeheader()
    for row in data:
        csvwrite.writerow(row)

为了保存空间,我想压缩它们。
使用gzip模块非常简单:

with gzip.open("foo.gz", "w") as csvfile :
    csvwrite = csv.DictWriter(csvfile, delimiter=',', fieldnames=fieldnames)
    csvwrite.writeheader()
    for row in data:
        csvwrite.writerow(row)

但我想要的文件在'zip'格式。
我尝试了zipfile模块,但我无法直接将文件写入zip存档。
相反,我必须将csv文件写入磁盘,使用以下代码将其压缩为zip文件,然后删除csv文件。

with ZipFile(zipfilepath, 'w') as zipfile:
    zipfile.write(csvfilepath, csvfilename, ZIP_DEFLATED)

如何将csv文件直接写入类似于gzip的压缩zip文件?

kqhtkvqz

kqhtkvqz1#

使用cStringIO.StringIO对象模拟文件:

with ZipFile(your_zip_file, 'w', ZIP_DEFLATED) as zip_file:
    string_buffer = StringIO()
    writer = csv.writer(string_buffer)

    # Write data using the writer object.

    zip_file.writestr(filename + '.csv', string_buffer.getvalue())
nfs0ujit

nfs0ujit2#

谢谢kroolik它做了一点修改。

with ZipFile(your_zip_file, 'w', ZIP_DEFLATED) as zip_file:
    string_buffer = StringIO()
    csvwriter = csv.DictWriter(string_buffer, delimiter=',', fieldnames=fieldnames)
    csvwrite.writeheader()
    for row in cdrdata:
        csvwrite.writerow(row)
    zip_file.writestr(filename + '.csv', string_buffer.getvalue())
rkttyhzu

rkttyhzu3#

让IOString存储内存中的每个字节可能会非常消耗内存。
根据zipfile模块文档,在创建一个ZipFile对象后,必须打开所有单独的文件,如下所示:

with ZipFile('spam.zip') as myzip:
    with myzip.open('eggs.txt') as myfile:
        print(myfile.read())

此示例也可用于write ...

相关问题