development

Python 목록을 CSV 파일로 작성

big-blog 2020. 6. 13. 09:26
반응형

Python 목록을 CSV 파일로 작성


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

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

파이썬의 내장 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)

당신은 사용할 수 있습니다 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)

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


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

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

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


매우 큰 목록에서 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")

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

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

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

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

If you don't want to import csv module for that, you can write a list of lists to a csv file using only Python built-ins

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

참고URL : https://stackoverflow.com/questions/14037540/writing-a-python-list-of-lists-to-a-csv-file

반응형