Pisanie listy list Pythona do pliku csv


168

Mam długą listę list w następującym formacie ---

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

tzn. wartości na liście są różnych typów - float, int, strings. jak zapisać je do pliku csv, aby mój wyjściowy plik csv wyglądał jak

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

Odpowiedzi:


299

Wbudowany moduł CSV Pythona może sobie z tym łatwo poradzić:

import csv

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

Zakłada się, że Twoja lista jest zdefiniowana tak a, jak w pytaniu. Możesz dostosować dokładny format wyjściowego pliku CSV za pomocą różnych parametrów opcjonalnych, csv.writer()zgodnie z dokumentacją na stronie referencyjnej biblioteki, do której link znajduje się powyżej.

Aktualizacja dla Pythona 3

import csv

with open("out.csv", "w", newline="") as f:
    writer = csv.writer(f)
    writer.writerows(a)

88
Aby zapewnić zgodność z Pythonem 3, usuń „b” z „wb”.
Vlad V

28
W Pythonie 3 - otwórz ('output.csv', 'w', newline = ''). Otrzymuję dodatkową linię, jeśli pominę parametr nowej linii. docs.python.org/3/library/csv.html#csv.writer
Spa

1
W pythonie3 musiałem użyć open ('output.csv', 'w', newline = "")
Tim Mottram

1
WOW, że błąd w Pythonie 3 jest bardzo nieprzydatny. Dzięki @vladV (wymagany jest obiekt przypominający bajty, a nie „str”). Z perspektywy czasu ma to sens, ale nie informuje, gdzie w ogóle patrzeć.
Rambatino

1
@tlalco, co prawdopodobnie oznacza, że ​​używasz Pythona 2, w takim przypadku powinieneś użyć pierwszego bloku kodu, a nie drugiego. (Oznacza to również, że powinieneś rozważyć przejście na Python 3.)
Amber

35

Możesz użyć 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)

Myślę, że nie powinienem używać, pandasjeśli wbudowana biblioteka csvmoże to zrobić.
Simin Jie

6
Lubię pandy, ponieważ jest potężny
dorbodwolf

9
pandy są potężne, jasne, ale nie zamierzam używać McLarena do jazdy do sąsiedniego sklepu na rogu.
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]])

oficjalne dokumenty: http://docs.python.org/2/library/csv.html


1
To dałoby moje +1, gdybyś mógł wyjaśnić swoją odpowiedź kilkoma komentarzami.
Burhan Khalid

3
writerownie przyjmuje wielu argumentów.
Amber

1
>>> w.writerow ("a", "b", "c") Traceback (ostatnie wywołanie ostatnie): Plik "<stdin>", wiersz 1, w <module> TypeError: writerow () przyjmuje dokładnie jeden argument (Podano 3)
Amber

@Amber jakiej wersji Pythona używasz?
Dmitry Zagorulkin

4
@Amber przepraszam. przegapiłem []
Dmitry Zagorulkin

11

Jeśli z jakiegoś powodu chciał zrobić to ręcznie (bez użycia modułu jak csv, pandas, numpyitd.):

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

Oczywiście toczenie własnej wersji może być podatne na błędy i nieefektywne ... dlatego zazwyczaj jest do tego odpowiedni moduł. Ale czasami pisanie własnych może pomóc ci zrozumieć, jak działają, a czasami jest to po prostu łatwiejsze.


11

Używanie csv.writer na mojej bardzo dużej liście zajęło sporo czasu. Postanowiłem użyć pand, było to szybsze i łatwiejsze do kontrolowania i zrozumienia:

 import pandas

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

Dobrze, że możesz coś zmienić, aby utworzyć lepszy plik 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")

5

Rozwiązanie Ambers działa również dobrze w przypadku tablic 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_)

4

Jeśli nie chcesz w tym celu importować csvmodułu, możesz zapisać listę list do pliku csv używając tylko wbudowanych Pythona

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

3

Pamiętaj, aby wskazać lineterinator='\n'podczas tworzenia pisarza; w przeciwnym razie dodatkowa pusta linia może zostać zapisana w pliku po każdym wierszu danych, gdy źródła danych pochodzą z innego pliku csv ...

Oto moje rozwiązanie:

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

3

Co powiesz na zrzucenie listy listy do marynaty i przywrócenie jej za pomocą modułu marynowania ? To całkiem wygodne.

>>> 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]]]
>>> 

0

Otrzymałem komunikat o błędzie, gdy postępowałem zgodnie z przykładami z parametrem nowej linii w funkcji csv.writer. Poniższy kod zadziałał dla mnie.

 with open(strFileName, "w") as f:
    writer = csv.writer(f, delimiter=',',  quoting=csv.QUOTE_MINIMAL)
    writer.writerows(result)
Korzystając z naszej strony potwierdzasz, że przeczytałeś(-aś) i rozumiesz nasze zasady używania plików cookie i zasady ochrony prywatności.
Licensed under cc by-sa 3.0 with attribution required.