Python grava em CSV linha por linha

101

Tenho dados que estão sendo acessados ​​via solicitação http e são devolvidos pelo servidor em formato separado por vírgulas, tenho o seguinte código:

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)

O conteúdo do texto é o seguinte:

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

Como posso salvar esses dados em um arquivo CSV. Eu sei que posso fazer algo ao longo das linhas a seguir para iterar linha por linha:

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

Mas não tenho certeza de como agora escrever corretamente cada linha em CSV

EDITAR ---> Obrigado pelo feedback, pois sugerido a solução era bastante simples e pode ser vista abaixo.

Solução:

import StringIO
s = StringIO.StringIO(text)
with open('fileName.csv', 'w') as f:
    for line in s:
        f.write(line)
Mustard Tiger
fonte
1
Já é um CSV, você só precisa escrever cada linha em um arquivo ...
icedwater
1
Não tenho certeza se você precisa mesmo da StringIOimportação, para ser honesto. Além disso, a solução no estado em que se encontra provavelmente não separa as linhas, pois f.write()não acrescenta novas linhas automaticamente.
icedwater de
@icedwater Eu entendo o que você está dizendo, mas executei o código acima e ele foi capaz de armazenar os dados corretamente em um arquivo csv.
Mustard Tiger
Consulte também: Como leio e escrevo arquivos CSV com Python?
Martin Thoma

Respostas:

175

Forma geral:

##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')

OU

Usando o escritor CSV:

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)

OU

Maneira mais simples:

f = open('csvfile.csv','w')
f.write('hi there\n') #Give your csv text here.
## Python will convert \n to os.linesep
f.close()
Ani Menon
fonte
16
para python 3, mude parawith open(<path to output_csv>, "w", newline='') as csv_file:
Khaled Hamed
1
Informações ausentes nesta linha for line in data:. Por favor, corrija isso. Obrigado.
Francisco Maria Calisto
@gsamaras A ideia era ajudar a comunidade, ao contrário de você editar e comentar que são inúteis.
Ani Menon
como acrescentar uma linha, se já houver algo no arquivo csv, usando a terceira solução?
Jürgen K.
3
@ JürgenK. Use 'a'(modo anexar) em vez de 'w'(modo de gravação).
Ani Menon
15

Você pode simplesmente gravar no arquivo como faria em qualquer arquivo normal.

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

Se apenas no caso, é uma lista de listas, você pode usar diretamente o csvmódulo integrado

import csv

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

Eu simplesmente escreveria cada linha em um arquivo, uma vez que já está em um formato CSV:

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

Não consigo me lembrar de como escrever linhas com quebras de linha no momento, embora: p

Além disso, você pode querer dar uma olhada esta resposta sobre write(), writelines()e '\n'.

água gelada
fonte
4

Para complementar as respostas anteriores, criei uma aula rápida para escrever em arquivos CSV. Torna mais fácil gerenciar e fechar arquivos abertos e obter consistência e código mais limpo se você tiver que lidar com vários arquivos.

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

Exemplo de uso:

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

Diverta-se

jrm
fonte
1

Que tal isso:

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

str.join () Retorna uma string que é a concatenação das strings em iterável. O separador entre os elementos é a string que fornece esse método.

Vlad Bezden
fonte