Làm cách nào để ghi dữ liệu vào định dạng csv dưới dạng chuỗi (không phải tệp)?

Question

Tôi muốn truyền dữ liệu như [1,2,'a','He said "what do you mean?"'] đến chuỗi được định dạng csv.

Thông thường, người dùng sẽ sử dụng csv.writer() vì điều này xử lý tất cả các trường hợp cạnh điên (thoát dấu phẩy, thoát dấu ngoặc kép, phương ngữ CSV, v.v.) Bắt giữ là csv.writer() sẽ xuất ra đối tượng tệp chứ không phải chuỗi.

giải pháp hiện tại của tôi là chức năng hơi hacky này:

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 

bất cứ ai có thể đưa ra một giải pháp thanh lịch mà vẫn xử lý các trường hợp cạnh tốt không?

Edit: Đây là cách tôi đã kết thúc làm việc đó:

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

Answer 1

Bạn có thể sử dụng StringIO thay vì riêng Dummy_Writer của bạn:

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

Ngoài ra còn có cStringIO, mà là một phiên bản nhanh hơn của StringIO lớp học.

Answer 2

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

Một số chi tiết cần phải được thay đổi một chút cho 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

Dưới đây là phiên bản mà làm việc cho utf-8. csvline2string cho chỉ một dòng, không xuống hàng cuối cùng, csv2string cho nhiều dòng, với linebreaks:

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

Tôi tìm thấy câu trả lời, tất cả trong tất cả, một chút bối rối. Đối với Python 2, cách sử dụng này hiệu quả với tôi:

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)