如何將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())

Answer 2

謝謝 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())