Eu tenho uma longa lista de listas do seguinte formulário ---
a = [[1.2,'abc',3],[1.2,'werew',4],........,[1.4,'qew',2]]
ou seja, os valores na lista são de tipos diferentes - float, int, strings.
1.2,abc,3
1.2,werew,4
.
.
.
1.4,qew,2
Eu tenho uma longa lista de listas do seguinte formulário ---
a = [[1.2,'abc',3],[1.2,'werew',4],........,[1.4,'qew',2]]
ou seja, os valores na lista são de tipos diferentes - float, int, strings.
1.2,abc,3
1.2,werew,4
.
.
.
1.4,qew,2
Respostas:
O módulo CSV embutido do Python pode lidar com isso facilmente:
import csv
with open("output.csv", "wb") as f:
writer = csv.writer(f)
writer.writerows(a)
Isso pressupõe que sua lista está definida como a
, como na sua pergunta. Você pode ajustar o formato exato do CSV de saída através dos vários parâmetros opcionais, csv.writer()
conforme documentado na página de referência da biblioteca vinculada acima.
Atualização para Python 3
import csv
with open("out.csv", "w", newline="") as f:
writer = csv.writer(f)
writer.writerows(a)
Você poderia usar 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
se a biblioteca interna csv
pode fazê-lo.
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]])
documentos oficiais: http://docs.python.org/2/library/csv.html
writerow
não aceita vários argumentos.
Se por qualquer motivo você queria fazê-lo manualmente (sem usar um módulo como 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')
Obviamente, lançar sua própria versão pode ser propenso a erros e ineficiente ... geralmente é por isso que existe um módulo para isso. Mas, às vezes, escrever o seu próprio texto pode ajudá-lo a entender como eles funcionam, e às vezes é mais fácil.
O uso de csv.writer na minha lista muito grande levou um bom tempo. Decidi usar pandas, era mais rápido e fácil de controlar e entender:
import pandas
yourlist = [[...],...,[...]]
pd = pandas.DataFrame(yourlist)
pd.to_csv("mylist.csv")
A parte boa é que você pode alterar algumas coisas para criar um arquivo csv melhor:
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")
A solução da Ambers também funciona bem para matrizes 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_)
Certifique-se de indicar lineterinator='\n'
quando criar o gravador; caso contrário, uma linha vazia extra poderá ser gravada no arquivo após cada linha de dados quando as fontes de dados forem de outro arquivo csv ...
Aqui está a minha solução:
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])
Que tal despejar a lista de lista em pickle e restaurá-la com o módulo pickle ? É bastante conveniente.
>>> 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]]]
>>>