Escribir una lista de listas de Python en un archivo csv

168

Tengo una larga lista de listas de la siguiente forma ---

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

es decir, los valores en la lista son de diferentes tipos: float, int, strings. ¿Cómo lo escribo en un archivo csv para que mi archivo csv de salida se vea como

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

fuente
2
Posible duplicado de Guardar resultados en archivo csv con Python
Martin Thoma

Respuestas:

299

El módulo CSV incorporado de Python puede manejar esto fácilmente:

import csv

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

Esto supone que su lista se define como a , como está en su pregunta. Puede ajustar el formato exacto del CSV de salida a través de los diversos parámetros opcionales csv.writer()según lo documentado en la página de referencia de la biblioteca vinculada anteriormente.

Actualización para Python 3

import csv

with open("out.csv", "w", newline="") as f:
    writer = csv.writer(f)
    writer.writerows(a)
Ámbar
fuente
88
Para compatibilidad con Python 3, elimine la "b" de "wb".
Vlad V
28
Con Python 3 - abierto ('output.csv', 'w', newline = ''). Obtengo una línea adicional si omito el parámetro de nueva línea. docs.python.org/3/library/csv.html#csv.writer
Spas del
1
En python3 tuve que usar open ('output.csv', 'w', newline = "")
Tim Mottram el
1
WOW, ese error de Python 3 es muy inútil. Gracias @vladV (se requiere un objeto similar a bytes, no 'str'). Tiene sentido en retrospectiva, pero no es informativo de dónde mirar.
Rambatino,
1
@tlalco que probablemente significa que está usando Python 2, en cuyo caso debe usar el primer bloque de código en lugar del segundo. (También significa que debería considerar cambiar a Python 3.)
Ámbar
35

Podrías 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)
Akavall
fuente
No creo que deba usarse pandassi la biblioteca integrada csvpuede hacerlo.
Simin Jie
66
me gustan los pandas porque es poderoso
dorbodwolf
9
los pandas son poderosos, claro, pero no voy a usar un McLaren para conducir a la tienda de la esquina de al lado.
MI Wright
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]])

documentos oficiales: http://docs.python.org/2/library/csv.html

Dmitry Zagorulkin
fuente
1
Esto obtendría mi +1 si pudiera explicar su respuesta con algunos comentarios.
Burhan Khalid
3
writerowNo toma múltiples argumentos.
Ámbar
1
>>> w.writerow ("a", "b", "c") Rastreo (última llamada más reciente): Archivo "<stdin>", línea 1, en <module> TypeError: writerow () toma exactamente un argumento (3 dados)
Ámbar
@Amber, ¿qué versión de Python usas?
Dmitry Zagorulkin
44
@ Ámbar discúlpame. me perdí []
Dmitry Zagorulkin
11

Si por cualquier razón usted quiere hacerlo manualmente (sin utilizar un módulo como csv, pandas, numpyetc.):

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

Por supuesto, lanzar su propia versión puede ser propenso a errores e ineficiente ... por eso generalmente hay un módulo para eso. Pero a veces escribir el suyo puede ayudarlo a comprender cómo funcionan, y a veces es más fácil.

tegan
fuente
11

El uso de csv.writer en mi gran lista tomó bastante tiempo. Decidí usar pandas, era más rápido y más fácil de controlar y comprender:

 import pandas

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

Lo bueno es que puedes cambiar algunas cosas para hacer un mejor archivo 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")
Dinidiniz
fuente
5

La solución de Ambers también funciona bien para arreglos 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_)
Semjon Mössinger
fuente
4

Si no desea importar un csvmódulo para eso, puede escribir una lista de listas en un archivo csv utilizando solo los elementos integrados de Python

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

Asegúrese de indicar lineterinator='\n'cuándo crear el escritor; de lo contrario, podría escribirse una línea vacía adicional en el archivo después de cada línea de datos cuando las fuentes de datos son de otro archivo csv ...

Aquí está mi solución:

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])
yashi wang
fuente
3

¿Qué tal volcar la lista de listas en pickle y restaurarla con el módulo pickle ? Es muy 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]]]
>>> 
Buena voluntad
fuente
0

Recibí un mensaje de error al seguir los ejemplos con un parámetro de nueva línea en la función csv.writer. El siguiente código funcionó para mí.

 with open(strFileName, "w") as f:
    writer = csv.writer(f, delimiter=',',  quoting=csv.QUOTE_MINIMAL)
    writer.writerows(result)
Jie
fuente