Bagaimana cara menulis data ke dalam format CSV sebagai string (bukan file)?

Saya ingin mentransmisikan data seperti [1,2,'a','He said "what do you mean?"']string berformat CSV.

Biasanya seseorang akan menggunakan csv.writer()untuk ini, karena menangani semua kasus tepi gila (pelolosan koma, pelolosan tanda kutip, dialek CSV, dll.) Tangkapannya adalah yang csv.writer()mengharapkan keluaran ke objek file, bukan ke string.

Solusi saya saat ini adalah fungsi yang agak hack ini:

def CSV_String_Writeline(data):
    class Dummy_Writer:
        def write(self,instring):
            self.outstring = instring.strip("\r\n")
    dw = Dummy_Writer()
    csv_w = csv.writer( dw )
    csv_w.writerow(data)
    return dw.outstring

Adakah yang bisa memberikan solusi yang lebih elegan yang masih menangani casing tepi dengan baik?

Sunting: Begini akhirnya saya melakukannya:

def csv2string(data):
    si = StringIO.StringIO()
    cw = csv.writer(si)
    cw.writerow(data)
    return si.getvalue().strip('\r\n')

Anda bisa menggunakan StringIOsebagai pengganti Anda sendiri Dummy_Writer:

Modul ini mengimplementasikan kelas seperti file StringIO,, yang membaca dan menulis buffer string (juga dikenal sebagai file memori).

Ada juga cStringIO, yang merupakan versi StringIOkelas yang lebih cepat .

Dengan Python 3:

>>> import io
>>> import csv
>>> output = io.StringIO()
>>> csvdata = [1,2,'a','He said "what do you mean?"',"Whoa!\nNewlines!"]
>>> writer = csv.writer(output, quoting=csv.QUOTE_NONNUMERIC)
>>> writer.writerow(csvdata)
59
>>> output.getvalue()
'1,2,"a","He said ""what do you mean?""","Whoa!\nNewlines!"\r\n'

Beberapa detail perlu sedikit diubah untuk Python 2:

>>> output = io.BytesIO()
>>> writer = csv.writer(output)
>>> writer.writerow(csvdata)
57L
>>> output.getvalue()
'1,2,a,"He said ""what do you mean?""","Whoa!\nNewlines!"\r\n'

Saya menemukan jawabannya, secara keseluruhan, agak membingungkan. Untuk Python 2, penggunaan ini berhasil untuk saya:

import csv, io

def csv2string(data):
    si = io.BytesIO()
    cw = csv.writer(si)
    cw.writerow(data)
    return si.getvalue().strip('\r\n')

data=[1,2,'a','He said "what do you mean?"']
print csv2string(data)

karena saya menggunakan ini cukup banyak untuk mengalirkan hasil secara asynchronous dari sanic kembali ke pengguna sebagai data csv saya menulis potongan berikut untuk Python 3 .

Cuplikan memungkinkan Anda menggunakan kembali buffer StringIo yang sama berulang kali.

import csv
from io import StringIO


class ArgsToCsv:
    def __init__(self, seperator=","):
        self.seperator = seperator
        self.buffer = StringIO()
        self.writer = csv.writer(self.buffer)

    def stringify(self, *args):
        self.writer.writerow(args)
        value = self.buffer.getvalue().strip("\r\n")
        self.buffer.seek(0)
        self.buffer.truncate(0)
        return value + "\n"

contoh:

csv_formatter = ArgsToCsv()

output += csv_formatter.stringify(
    10,
    """
    lol i have some pretty
    "freaky"
    strings right here \' yo!
    """,
    [10, 20, 30],
)

Lihat penggunaan lebih lanjut di github gist: source and test

import csv
from StringIO import StringIO
with open('file.csv') as file:
    file = file.read()

stream = StringIO(file)

csv_file = csv.DictReader(stream)

Ini versi yang berfungsi untuk utf-8. csvline2string hanya untuk satu baris, tanpa linebreak di akhir, csv2string untuk banyak baris, dengan linebreak:

import csv, io

def csvline2string(one_line_of_data):
    si = BytesIO.StringIO()
    cw = csv.writer(si)
    cw.writerow(one_line_of_data)
    return si.getvalue().strip('\r\n')

def csv2string(data):
    si = BytesIO.StringIO()
    cw = csv.writer(si)
    for one_line_of_data in data:
        cw.writerow(one_line_of_data)
    return si.getvalue()