it-roy-ru.com

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

У меня есть данные, доступ к которым осуществляется через 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)
49
Mustard Tiger

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

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

OR

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

OR

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

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

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

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
Vishwa

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

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

Я не могу вспомнить, как писать строки с переносами строк, хотя: p

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

7
icedwater

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

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

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

1
Vlad Bezden