python CSV 파일 쓰기

python CSV 파일 쓰기

CSV 파일을 쓰기 위해서는 .csv 파일을 쓰기모드로 오픈하고 파일객체를 csv.writer(파일객체) 에 넣으면 됩니다. CSV writer는 writerow() 라는 메서드를 통해 list 데이타를 한 라인 추가하게 됩니다. 윈도우즈의 경우 csv 모듈에서 데이타를 쓸 때 각 라인 뒤에 빈 라인이 추가되는 문제가 있는데, 이를 없애기 위해 (파이썬 3 에서) 파일을 open 할 때 newline='' 와 같은 옵션을 지정합니다 (주: 파이썬 2의 경우는 newline 옵션 없이 바이너리 모드로 오픈할 것). 아래 예제는 Book1.csv 라는 CSV 파일에 2개 라인을 추가하는 예입니다.

#-*- coding: utf-8 -*-
'''
Created on 2018. 12. 8.

@author: Lee
'''
import csv    
f = open('Book1.csv', 'w', encoding='utf-8', newline='')
wr = csv.writer(f)
wr.writerow([1, "홍길동", "남자"])
wr.writerow([2, "김영희", "여자"])
f.close()