python - 如何将 csv 文件直接压缩成 zip 存档？

Question

我正在使用以下代码动态生成许多 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？

Answer 1

使用 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())