Python scrive in CSV riga per riga


102

Ho dati a cui si accede tramite richiesta http e vengono restituiti dal server in un formato separato da virgole, ho il seguente codice:

site= 'www.example.com'
hdr = {'User-Agent': 'Mozilla/5.0'}
req = urllib2.Request(site,headers=hdr)
page = urllib2.urlopen(req)
soup = BeautifulSoup(page)
soup = soup.get_text()
text=str(soup)

Il contenuto del testo è il seguente:

april,2,5,7
may,3,5,8
june,4,7,3
july,5,6,9

Come posso salvare questi dati in un file CSV. So di poter fare qualcosa sulla falsariga di quanto segue per iterare riga per riga:

import StringIO
s = StringIO.StringIO(text)
for line in s:

Ma non sono sicuro di come scrivere ora correttamente ogni riga in CSV

EDIT ---> Grazie per il feedback come suggerito, la soluzione era piuttosto semplice e può essere vista di seguito.

Soluzione:

import StringIO
s = StringIO.StringIO(text)
with open('fileName.csv', 'w') as f:
    for line in s:
        f.write(line)

1
È già un CSV, devi solo scrivere ogni riga su un file ...
icedwater

1
Non sono nemmeno sicuro che tu abbia bisogno StringIOdell'importazione, ad essere onesti. Inoltre, la soluzione così com'è probabilmente non separa le righe, poiché f.write()non aggiunge automaticamente le nuove righe.
icedwater

@icedwater Capisco cosa stai dicendo, ma ho eseguito il codice sopra ed è stato in grado di memorizzare correttamente i dati in un file csv.
Mustard Tiger

Risposte:


176

Modo generale:

##text=List of strings to be written to file
with open('csvfile.csv','wb') as file:
    for line in text:
        file.write(line)
        file.write('\n')

O

Utilizzando CSV writer:

import csv
with open(<path to output_csv>, "wb") as csv_file:
        writer = csv.writer(csv_file, delimiter=',')
        for line in data:
            writer.writerow(line)

O

Modo più semplice:

f = open('csvfile.csv','w')
f.write('hi there\n') #Give your csv text here.
## Python will convert \n to os.linesep
f.close()

17
per Python 3, cambialo inwith open(<path to output_csv>, "w", newline='') as csv_file:
Khaled Hamed

1
Informazioni mancanti su questa riga for line in data:. Per favore aggiustalo. Grazie.
Francisco Maria Calisto

@gsamaras L'idea era di aiutare la comunità, diversamente da te che modifichi e commenti che sono inutili.
Ani Menon

come aggiungere una riga, se c'è già qualcosa nel file csv, usando la terza soluzione?
Jürgen K.

3
@ JürgenK. Utilizzare 'a'(modalità aggiunta) al posto di 'w'(modalità scrittura).
Ani Menon

15

Puoi semplicemente scrivere nel file come faresti con qualsiasi file normale.

with open('csvfile.csv','wb') as file:
    for l in text:
        file.write(l)
        file.write('\n')

Se per ogni evenienza si tratta di un elenco di elenchi, potresti utilizzare direttamente il csvmodulo integrato

import csv

with open("csvfile.csv", "wb") as file:
    writer = csv.writer(file)
    writer.writerows(text)

8

Vorrei semplicemente scrivere ogni riga in un file, poiché è già in un formato CSV:

write_file = "output.csv"
with open(write_file, "w") as output:
    for line in text:
        output.write(line + '\n')

Al momento, però, non ricordo come scrivere righe con interruzioni di riga: p

Inoltre, come si potrebbe dare un'occhiata a questa risposta su write(), writelines()e '\n'.


4

Per completare le risposte precedenti, ho creato una lezione veloce per scrivere su file CSV. Rende più facile gestire e chiudere i file aperti e ottenere coerenza e codice più pulito se devi gestire più file.

class CSVWriter():

    filename = None
    fp = None
    writer = None

    def __init__(self, filename):
        self.filename = filename
        self.fp = open(self.filename, 'w', encoding='utf8')
        self.writer = csv.writer(self.fp, delimiter=';', quotechar='"', quoting=csv.QUOTE_ALL, lineterminator='\n')

    def close(self):
        self.fp.close()

    def write(self, elems):
        self.writer.writerow(elems)

    def size(self):
        return os.path.getsize(self.filename)

    def fname(self):
        return self.filename

Utilizzo di esempio:

mycsv = CSVWriter('/tmp/test.csv')
mycsv.write((12,'green','apples'))
mycsv.write((7,'yellow','bananas'))
mycsv.close()
print("Written %d bytes to %s" % (mycsv.size(), mycsv.fname()))

Divertiti


1

Che dire di questo:

with open("your_csv_file.csv", "w") as f:
    f.write("\n".join(text))

str.join () Restituisce una stringa che è la concatenazione delle stringhe in iterable. Il separatore tra gli elementi è la stringa che fornisce questo metodo.

Utilizzando il nostro sito, riconosci di aver letto e compreso le nostre Informativa sui cookie e Informativa sulla privacy.
Licensed under cc by-sa 3.0 with attribution required.