Запись списка списков Python в файл csv

у меня есть длинный список списков в следующей форме ---

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

7 ответов


встроенный Python CSV модуль может справиться с этим легко:

import csv

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

Это предполагает, что ваш список определяется как a, как в вашем вопросе. Вы можете настроить точный формат выходного CSV с помощью различных дополнительных параметров в csv.writer() как описано в справочной странице библиотеки, связанной выше.


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


можно использовать 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)

использование csv.писателю в моем очень большом списке потребовалось немало времени. Я решил использовать панд, это было быстрее и проще контролировать и понимать:

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

если по какой-либо причине вы хотели сделать это вручную (без использования модуля, такого как csv,pandas,numpy etc.):

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

конечно, прокатка собственной версии может быть подвержена ошибкам и неэффективна ... обычно для этого есть модуль. Но иногда написание собственных может помочь вам понять, как они работают, а иногда это просто проще.


решение 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_)

указывать 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])