Come posso scrivere i dati in formato csv come stringa (non file)?

Question

Voglio trasmettere dati come [1,2,'a','He said "what do you mean?"'] a una stringa in formato csv.

Normalmente si potrebbe usare csv.writer() per questo, perché gestisce tutti i casi limite folli (virgola fuga, segno di citazione in fuga, dialetti CSV, ecc) Il problema è che csv.writer() prevede di output in un oggetto file, non a un stringa.

mia soluzione attuale è questa funzione un po 'hacky:

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 

Qualcuno può dare una soluzione più elegante che gestisce ancora i casi limite bene?

Edit: Ecco come ho finito per farlo:

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

Answer 1

Si potrebbe utilizzare StringIO al posto della propria Dummy_Writer:

This module implements a file-like class, StringIO , that reads and writes a string buffer (also known as memory files).

C'è anche cStringIO, che è una versione più veloce del StringIO classe.

Answer 2

In 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' 

Alcuni dettagli devono essere cambiato un po 'per 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' 

Answer 3

Ecco la versione che funziona per utf-8. csvline2string per una sola riga, senza interruzioni di linea, alla fine, csv2string per molte linee, con interruzioni di linea:

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

Answer 4

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

stream = StringIO(file) 

csv_file = csv.DictReader(stream) 

Answer 5

ho trovato le risposte, tutto sommato, un po 'di confusione. Per Python 2, questo utilizzo ha funzionato per me:

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)