Python пише в CSV рядок за рядком


101

У мене є дані, доступ до яких здійснюється через 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, вам просто потрібно записати кожен рядок у файл ...
крижана вода

1
Я не впевнений, що вам навіть потрібен StringIOімпорт, якщо чесно. Крім того, рішення як є, ймовірно, не відокремлює рядки, оскільки f.write()не додає нові рядки автоматично.
крижана вода

@icedwater Я розумію, що ви говорите, але я запустив код вище, і він зміг належним чином зберегти дані у файл CSV.
Гірчичний тигр

Відповіді:


175

Загальний спосіб:

##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-записувача:

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

16
для python 3, змініть його наwith open(<path to output_csv>, "w", newline='') as csv_file:
Халед Хамед

1
У цьому рядку відсутня інформація for line in data:. Будь ласка, виправте це. Дякую.
Francisco Maria Calisto

@gsamaras Ідея полягала в тому, щоб допомогти спільноті, на відміну від редагування та коментування, які марні.
Ані Менон,

як додати рядок, якщо в 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')

На даний момент я не пам’ятаю, як писати рядки з розривами рядків: с

Крім того , ви хотіли б поглянути на цей відповідь про 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()))

Весело


1

Як що до цього:

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

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

Використовуючи наш веб-сайт, ви визнаєте, що прочитали та зрозуміли наші Політику щодо файлів cookie та Політику конфіденційності.
Licensed under cc by-sa 3.0 with attribution required.