Запись списка списков Python в файл csv
У меня есть длинный список списков следующего вида ---
a = [[1.2,'abc',3],[1.2,'werew',4],........,[1.4,'qew',2]]
то есть. значения в списке имеют разные типы - float, int, string.How я пишу его в файл csv, чтобы мой выходной файл csv выглядел как
1.2,abc,3
1.2,werew,4
.
.
.
1.4,qew,2
Ответы
Ответ 1
Встроенный CSV-модуль Python может легко справиться с этим:
import csv
with open("output.csv", "wb") as f:
writer = csv.writer(f)
writer.writerows(a)
Это предполагает ваш список определяется как, так как это в вашем вопросе. a
Вы можете настроить точный формат выходного CSV с помощью различных необязательных параметров для csv.writer()
как csv.writer()
на справочной странице библиотеки, указанной выше.
Обновление для Python 3
import csv
with open("out.csv", "w", newline="") as f:
writer = csv.writer(f)
writer.writerows(a)
Ответ 2
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
Ответ 3
Вы можете использовать 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)
Ответ 4
Если по какой-либо причине вы хотели сделать это вручную (без использования модуля типа csv
, pandas
, numpy
и т.д.):
with open('myfile.csv','w') as f:
for sublist in mylist:
for item in sublist:
f.write(item + ',')
f.write('\n')
Конечно, запуск вашей собственной версии может быть подвержен ошибкам и неэффективен... что обычно для этого есть модуль для этого. Но иногда писать свои собственные могут помочь вам понять, как они работают, а иногда и проще.
Ответ 5
Использование csv.writer в моем очень большом списке заняло довольно много времени. Я решил использовать pandas, быстрее и легче управлять и понимать:
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")
Ответ 6
Решение Ambers также хорошо работает для массивов numpy:
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_)
Ответ 7
Обязательно укажите 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])
Ответ 8
Как насчет выгрузки списка списков в pickle и восстановления его с помощью модуля pickle? Это довольно удобно.
>>> 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]]]
>>>
Ответ 9
Я отмечаю, что для преобразования вашего списка в фрейм данных Pandas может потребоваться много памяти. Я столкнулся с ошибками памяти, пытаясь записать списки в CSV файлы путем преобразования в dataframes. те ответы выше, которые предлагают записывать списки непосредственно в CSV файлы, являются лучшими ответами с этой точки зрения.
Ответ 10
Если вы не хотите импортировать модуль 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))