Écriture D'une liste Python de listes dans un fichier csv
, j'ai une longue liste de listes de la forme suivante ---
a = [[1.2,'abc',3],[1.2,'werew',4],........,[1.4,'qew',2]]
C'est-à-dire que les valeurs de la liste sont de différents types -- float,int, strings.Comment puis-je l'écrire dans un fichier csv pour que mon fichier csv de sortie ressemble à
1.2,abc,3
1.2,werew,4
.
.
.
1.4,qew,2
7 réponses
Le module CSV intégré de Python peut gérer cela facilement:
import csv
with open("output.csv", "wb") as f:
writer = csv.writer(f)
writer.writerows(a)
Cela suppose que votre liste est définie comme a, comme il est dans votre question. Vous pouvez modifier le format exact de la sortie CSV via les différents paramètres optionnels à csv.writer() comme documenté dans la page de référence de la bibliothèque liée ci-dessus.
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]])
Documents officiels: http://docs.python.org/2/library/csv.html
Vous pouvez utiliser 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)
En utilisant csv.écrivain dans ma très grande liste a pris un certain temps. J'ai décidé d'utiliser des pandas, c'était plus rapide et plus facile à contrôler et à comprendre:
import pandas
yourlist = [[...],...,[...]]
pd = pandas.DataFrame(yourlist)
pd.to_csv("mylist.csv")
La bonne partie vous pouvez changer quelque chose pour faire un meilleur fichier 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")
Si pour une raison quelconque vous vouliez le faire manuellement (sans utiliser un module comme 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')
Bien sûr, rouler votre propre version peut être sujet aux erreurs et inefficace ... c'est généralement pourquoi il y a un module pour cela. Mais parfois, écrire le vôtre peut vous aider à comprendre comment ils fonctionnent, et parfois c'est juste plus facile.
La solution D'Ambers fonctionne également bien pour les tableaux 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_)
Assurez-vous d'indiquer lineterinator='\n' lors de la création de l'auteur; sinon, une ligne vide supplémentaire peut être écrite dans le fichier après chaque ligne de données lorsque les sources de données proviennent d'un autre fichier csv...
Voici ma solution:
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])