Python - 在内存中生成 csv 文件,然后将其数据编码为 base64?

Python - generate csv file in memory and then encode its data into base64?

我需要在内存中生成类似数据的csv文件,然后将其编码为base64,以便保存。所以基本上我不想为此在硬盘中创建文件。现在我通过创建 csv 文件然后对其数据进行编码、保存然后简单地删除 csv 文件(因为不再需要它)来解决这个问题。但是有没有一种方法可以跳过文件创建,而是以相同的方式保存数据?我的意思是数据将用于使用 base64 再次打开 csv 文件。

import base64
import csv
from StringIO import StringIO
import os

def test_binary(self):
    mylist = [['a', 'b'], ['c', 'd']]
    with open("test.csv", "wb") as f:
        writer = csv.writer(f)
        writer.writerows(mylist)   

    myfile = open('test.csv', 'r')
    stream = StringIO(myfile.read())

    encoded = base64.b64encode(stream.getvalue())

    self.test = encoded
    myfile.close()
    os.remove('test.csv')        

如果我没理解错的话,您的最终目标是将列表转换为 .csv 格式的字符串,并对其进行 base64 编码。如果是这种情况,您根本不需要创建 .csv 文件。

import base64

mylist = [['a', 'b'], ['c', 'd']]

# Creating the csv formatted string
csv_line_length = len(max(mylist,key=len))
csv_string = ''
for row in mylist:
    temp_row = ['"' + col + '"' for col in row]
    while len(temp_row) < csv_line_length:
      temp_row.append([])
    csv_string += ','.join(temp_row) + '\n'
# Encoding the string to base64
encoded = base64.b64encode(csv_string.encode('utf-8'))

您可以将 StringIO() 直接传递给 csv 编写器:

>>> import base64
>>> import csv
>>> from io import StringIO
>>> mylist = [['a', 'b'], ['c', 'd']]
>>> f = StringIO()
>>> csv.writer(f).writerows(mylist)
>>> base64.b64encode(f.getvalue().encode())
b'YSxiDQpjLGQNCg=='

Python 2

>>> import base64
>>> import csv
>>> from StringIO import StringIO
>>> mylist = [['a', 'b'], ['c', 'd']]
>>> f = StringIO()
>>> csv.writer(f).writerows(mylist)
>>> base64.b64encode(f.getvalue())
'YSxiDQpjLGQNCg=='