Написання списку Python у список файлів csv


168

У мене довгий список списків такої форми ---

a = [[1.2,'abc',3],[1.2,'werew',4],........,[1.4,'qew',2]]

тобто значення у списку бувають різних типів - float, int, strings. Як я записую його у файл csv, щоб мій вихідний файл csv виглядав як

1.2,abc,3
1.2,werew,4
.
.
.
1.4,qew,2

Відповіді:


299

Вбудований модуль CSV Python може легко впоратися з цим:

import csv

with open("output.csv", "wb") as f:
    writer = csv.writer(f)
    writer.writerows(a)

Це передбачає, що ваш список визначений як a, як і у вашому запитанні. Ви можете налаштувати точний формат вихідного CSV через різні необов'язкові параметри, csv.writer()як це задокументовано на довідковій сторінці бібліотеки, пов’язаній вище.

Оновлення для Python 3

import csv

with open("out.csv", "w", newline="") as f:
    writer = csv.writer(f)
    writer.writerows(a)

88
Для сумісності Python 3 видаліть "b" з "wb".
Влад V

28
З Python 3 - відкрити ('output.csv', 'w', newline = ''). Я отримую додатковий рядок, якщо опускаю параметр newline. docs.python.org/3/library/csv.html#csv.writer
Спас

1
У python3 мені довелося використовувати open ('output.csv', 'w', newline = "")
Тім Моттрам

1
ЯК, що помилка python 3 дуже не допомагає. Спасибі @vladV (потрібний байт-об’єкт, а не "str"). Це ніби має сенс заднього огляду, але не є інформативним, де його взагалі дивитись.
Рамбатіно

1
@tlalco, мабуть, означає, що ви використовуєте Python 2, і в цьому випадку ви повинні використовувати перший код коду, а не другий. (Це також означає, що вам слід подумати про перехід на Python 3.)
Бурштин,

35

Ви можете використовувати pandas:

In [1]: import pandas as pd

In [2]: a = [[1.2,'abc',3],[1.2,'werew',4],[1.4,'qew',2]]

In [3]: my_df = pd.DataFrame(a)

In [4]: my_df.to_csv('my_csv.csv', index=False, header=False)

Я не думаю, що слід використовувати, pandasякщо вбудована бібліотека csvможе це зробити.
Simin Jie

6
Мені подобаються панди, тому що його потужний
дорбодолюб

9
Панди є потужними, звичайно, але я не збираюся використовувати McLaren, щоб їхати до кутового магазину поруч.
М. І. Райт

30
import csv
with open(file_path, 'a') as outcsv:   
    #configure writer to write standard csv file
    writer = csv.writer(outcsv, delimiter=',', quotechar='|', quoting=csv.QUOTE_MINIMAL, lineterminator='\n')
    writer.writerow(['number', 'text', 'number'])
    for item in list:
        #Write item to outcsv
        writer.writerow([item[0], item[1], item[2]])

офіційні документи: http://docs.python.org/2/library/csv.html


1
Це отримає мій +1, якщо ви можете пояснити свою відповідь за допомогою деяких коментарів.
Бурхан Халід

3
writerowне бере декількох аргументів.
Бурштин

1
>>> w.writerow ("a", "b", "c") Traceback (останній дзвінок останній): Файл "<stdin>", рядок 1, в <module> TypeError: writerow () бере рівно один аргумент (3 надано)
Бурштин

@Amber яку версію python ви використовуєте?
Дмитро Загорулкін

4
@ Amber вибач мене. я пропустив []
Дмитро Загорулкін

11

Якщо з якоїсь - небудь причини ви хочете , щоб зробити це вручну (без використання модуля , як csv, pandas, і numpyт.д.):

with open('myfile.csv','w') as f:
    for sublist in mylist:
        for item in sublist:
            f.write(item + ',')
        f.write('\n')

Звичайно, прокрутка власної версії може бути схильною до помилок та неефективною ... саме тому для цього є модуль. Але іноді написання власного може допомогти вам зрозуміти, як вони працюють, а іноді це просто простіше.


11

Використання csv.writer у моєму дуже великому списку зайняло досить багато часу. Я вирішив використовувати панди, це було швидше і простіше контролювати і розуміти:

 import pandas

 yourlist = [[...],...,[...]]
 pd = pandas.DataFrame(yourlist)
 pd.to_csv("mylist.csv")

Хороша частина ви можете змінити щось, щоб зробити кращий файл CSV:

 yourlist = [[...],...,[...]]
 columns = ["abcd","bcde","cdef"] #a csv with 3 columns
 index = [i[0] for i in yourlist] #first element of every list in yourlist
 not_index_list = [i[1:] for i in yourlist]
 pd = pandas.DataFrame(not_index_list, columns = columns, index = index)

 #Now you have a csv with columns and index:
 pd.to_csv("mylist.csv")

5

Рішення бурштину також добре працює для нумерованих масивів:

from pylab import *
import csv

array_=arange(0,10,1)
list_=[array_,array_*2,array_*3]
with open("output.csv", "wb") as f:
    writer = csv.writer(f)
    writer.writerows(list_)

4

Якщо ви не хочете імпортувати csvмодуль для цього, ви можете написати список списків у файл csv, використовуючи лише вбудовані Python

with open("output.csv", "w") as f:
    for row in a:
        f.write("%s\n" % ','.join(str(col) for col in row))

3

Обов’язково вкажіть, lineterinator='\n'коли створюєте письменника; в іншому випадку додатковий порожній рядок може бути записаний у файл після кожного рядка даних, коли джерела даних є з іншого файлу CSV ...

Ось моє рішення:

with open('csvfile', 'a') as csvfile:
    spamwriter = csv.writer(csvfile, delimiter='    ',quotechar='|', quoting=csv.QUOTE_MINIMAL, lineterminator='\n')
for i in range(0, len(data)):
    spamwriter.writerow(data[i])

3

Як щодо скидання списку в соління та відновлення його за допомогою модуля соління ? Це досить зручно.

>>> import pickle
>>> 
>>> mylist = [1, 'foo', 'bar', {1, 2, 3}, [ [1,4,2,6], [3,6,0,10]]]
>>> with open('mylist', 'wb') as f:
...     pickle.dump(mylist, f) 


>>> with open('mylist', 'rb') as f:
...      mylist = pickle.load(f)
>>> mylist
[1, 'foo', 'bar', {1, 2, 3}, [[1, 4, 2, 6], [3, 6, 0, 10]]]
>>> 

0

Я отримав повідомлення про помилку , коли після прикладів з нового рядка параметра у функції csv.writer. Наступний код працював для мене.

 with open(strFileName, "w") as f:
    writer = csv.writer(f, delimiter=',',  quoting=csv.QUOTE_MINIMAL)
    writer.writerows(result)
Використовуючи наш веб-сайт, ви визнаєте, що прочитали та зрозуміли наші Політику щодо файлів cookie та Політику конфіденційності.
Licensed under cc by-sa 3.0 with attribution required.