Python писать в CSV построчно

102

У меня есть данные, к которым осуществляется доступ через http-запрос и которые отправляются сервером в формате, разделенном запятыми, у меня есть следующий код:

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)

Содержание текста следующее:

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

Как сохранить эти данные в файл CSV. Я знаю, что могу сделать что-то вроде следующего, чтобы перебирать строку за строкой:

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

Но я не уверен, как теперь правильно записывать каждую строку в CSV

РЕДАКТИРОВАТЬ ---> Спасибо за отзыв, поскольку решение было довольно простым и его можно увидеть ниже.

Решение:

import StringIO
s = StringIO.StringIO(text)
with open('fileName.csv', 'w') as f:
    for line in s:
        f.write(line)
Горчичный тигр
источник
1
Это уже CSV, вам просто нужно записать каждую строчку в файл ...
icedwater
1
Я не уверен, что вам вообще нужен StringIOимпорт, если честно. Кроме того, решение как есть, вероятно, не разделяет строки, так как f.write()не добавляет новые строки автоматически.
Iceedwater
@icedwater Я понимаю, что вы говорите, но я запустил приведенный выше код, и он смог правильно сохранить данные в файл csv.
Горчичный тигр

Ответы:

176

Общий способ:

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

ИЛИ

Использование 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)

ИЛИ

Самый простой способ:

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
для python 3 измените его наwith open(<path to output_csv>, "w", newline='') as csv_file:
Khaled Hamed
1
Отсутствует информация в этой строке for line in data:. Пожалуйста, исправьте это. Спасибо.
Франсиско Мария Калисто
@gsamaras Идея заключалась в том, чтобы помочь сообществу, в отличие от того, что вы редактируете и комментируете, которые бесполезны.
Ani Menon
как добавить строку, если в csv-файле уже что-то есть, с помощью третьего решения?
Юрген К.
3
@ JürgenK. Используйте 'a'(режим добавления) вместо 'w'(режим записи).
Ани Менон
15

Вы можете просто записать в файл, как если бы вы записали любой обычный файл.

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

Если на всякий случай это список списков, вы можете напрямую использовать встроенный csvмодуль

import csv

with open("csvfile.csv", "wb") as file:
    writer = csv.writer(file)
    writer.writerows(text)
Вишва
источник
8

Я бы просто записал каждую строку в файл, поскольку он уже в формате CSV:

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

Я не могу вспомнить, как писать строки с переносом строки в данный момент: p

Кроме того , вы хотели бы взглянуть на этот ответ о write(), writelines()и '\n'.

ледяная вода
источник
4

Чтобы дополнить предыдущие ответы, я создал быстрый класс для записи в файлы CSV. Это упрощает управление открытыми файлами и их закрытие, а также обеспечивает согласованность и более чистый код, если вам приходится иметь дело с несколькими файлами.

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

Пример использования:

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

Радоваться, веселиться

jrm
источник
1

Как насчет этого:

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

str.join () Возвращает строку, которая является конкатенацией строк в итерабельном. Разделителем между элементами является строка, обеспечивающая этот метод.

Влад Безден
источник