Python 목록을 CSV 파일로 작성


168

다음과 같은 형식의 긴 목록이 있습니다 ---

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

답변:


299

파이썬의 내장 CSV 모듈 은 이것을 쉽게 처리 할 수 ​​있습니다 :

import csv

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

이것은 목록이 다음과 같이 정의되었다고 가정합니다. a 귀하의 질문과 같이로 . csv.writer()위에 링크 된 라이브러리 참조 페이지에 설명 된대로 다양한 선택적 매개 변수를 통해 출력 CSV의 정확한 형식을 조정할 수 있습니다 .

파이썬 3 업데이트

import csv

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

88
Python 3 호환성을 위해 "wb"에서 "b"를 제거하십시오.
Vlad V

28
파이썬 3에서-open ( 'output.csv', 'w', newline = ''). 줄 바꿈 매개 변수를 생략하면 추가 줄이 나타납니다. docs.python.org/3/library/csv.html#csv.writer
스파

1
python3에서는 open ( 'output.csv', 'w', newline = "")을 사용해야했습니다
Tim Mottram

1
파이썬 3 오류는 매우 도움이되지 않습니다. @vladV에게 감사드립니다 ( 'str'이 아닌 바이트와 같은 객체가 필요합니다). 뒤늦게 이해하는 것이 좋지만 어디를보아야하는지에 대한 정보는 없습니다.
Rambatino

1
@tlalco는 아마도 파이썬 2를 사용하고 있음을 의미합니다.이 경우 두 번째 코드가 아닌 첫 번째 코드 블록을 사용해야합니다. (또한 파이썬 3으로 전환하는 것을 고려해야합니다.)
Amber

35

당신은 사용할 수 있습니다 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내장 라이브러리로 csv할 수 있다면 사용해야한다고 생각하지 않습니다 .
Simin Jie

6
나는 판다를 좋아한다
dorbodwolf

9
팬더는 강력하지만 확실히 옆집에 모퉁이 상점으로 운전하기 위해 McLaren을 사용하지는 않을 것입니다.
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]])

공식 문서 : http://docs.python.org/2/library/csv.html


1
댓글로 답변을 설명 할 수 있다면 +1이됩니다.
Burhan Khalid

3
writerow여러 개의 인수를 사용하지 않습니다.
Amber

1
>>> w.writerow ( "a", "b", "c") 역 추적 (최근 호출 마지막) : <module>의 파일 "<stdin>", 줄 1, TypeError : writerow ()는 정확히 하나의 인수를 사용합니다. (3 given)
Amber

@Amber 어떤 버전의 파이썬을 사용하십니까?
Dmitry Zagorulkin

4
@ 앰버 실례합니다. [] 그리워
드미트리 Zagorulkin

11

어떤 이유로 당신은 (같은 모듈을 사용하지 않고 수동으로 할 싶었다면 csv, pandas, numpy등) :

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

물론, 자신의 버전을 롤링하는 것은 오류가 발생하기 쉽고 비효율적 일 수 있습니다. 그러나 때때로 자신의 글을 작성하면 그들이 어떻게 작동하는지 이해하는 데 도움이 될 수 있으며 때로는 더 쉽습니다.


11

매우 큰 목록에서 csv.writer를 사용하는 데 상당한 시간이 걸렸습니다. 나는 팬더를 사용하기로 결정했는데, 더 빠르고 쉽게 제어하고 이해할 수있었습니다.

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

5

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

4

이를 csv위해 모듈 을 가져 오지 않으려면 Python 내장 기능 만 사용하여 csv 파일에 목록 목록을 작성할 수 있습니다

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

3

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

3

리스트리스트를 피클에 덤프하고 피클 모듈로 복원하는 것은 어떻습니까? 매우 편리합니다.

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

csv.writer 함수에 개행 매개 변수가있는 예제를 수행 할 때 오류 메시지가 나타납니다 . 다음 코드는 저에게 효과적이었습니다.

 with open(strFileName, "w") as f:
    writer = csv.writer(f, delimiter=',',  quoting=csv.QUOTE_MINIMAL)
    writer.writerows(result)
당사 사이트를 사용함과 동시에 당사의 쿠키 정책개인정보 보호정책을 읽고 이해하였음을 인정하는 것으로 간주합니다.
Licensed under cc by-sa 3.0 with attribution required.