파이썬기초 - CSV 파일 읽고 쓰기

CSV는 Comma Separated Values의 약자로 이름에서 알 수 있듯이 콤마로 구분된 텍스트 데이터를 가지고 있는 데이터 파일을 의미합니다.

확장자는 .csv로서 오래전부터 스프레드시트나 데이터베이스 소프트웨어에서 많이 사용되어져 왔으며, 엑셀에서도 열리기 때문에 많이 사용되어 지고 있습니다.

 

cvs화일을 사용하기 위해서는 우선 csv 메서드를 import하고, csv 파일을 열어야 합니다. 

그리고, csv파일을 읽고 쓰기 위해서는 아래와 같이 csv.reader()와 csv.writer()를 사용하면 됩니다.

그 외의 부분은 이전에 배웠던 파일읽기 쓰기와 유사하기 때문에 이해하는데 어렵지 않을 것이라 생각됩니다. 

 

import csv

 

f = open(‘data.csv’, ‘r’)  #읽기모드로 csv file open

rdr = csv.reader(f)

for line in rdr:            #csv file을 한줄씩 리스트형식([])으로 읽어들인다..

print(line)                

f.close()

 

csv file을 line이라는 이름의 리스트형식으로 한줄씩 읽었기 때문에 첫번째 열만 출력하기 위해서는 print(line[0])을 해주면 됩니다.

 

다음으로는 csv file을 만드는 방법인데요. 한글을 사용하기 위해서는 아래와 같이 encoding = 'utf-8'을 추가해 줘야 합니다.

     

import csv

 

f = open(‘data.csv’, ‘w’, encoding = ‘utf-8)

wr = csv.writer(f)

 

wr.writerrow([1, ‘오늘’, '2019-03-05’])

wr.writerrow([2, ‘어제, '2019-03-04’])

f.close()

 

여기서 writerrow 괄호안에는 리스트를 넣어주면 되며, 데이터가 많을 경우는 for문등을 사용하면 되겠죠.

 

여기까지가 파이썬을 이용해서 파일을 읽고 쓰는 가장 기본적인 부분을 배웠는데요.

이 정도만 알고 필요한 부분을 그 때 그 때 찾아보면 크게 문제가 없습니다.

 

파일을 읽고 쓸 때는 우선 파일을 열고 사용이 끝났으면 그 파일을 닫아줘야 에러가 발생하지 않는데요.

아래와 같이 with open문을 사용하면 with open이 끝나면 자동으로 파일이 닫히기 때문에 이 부분도 알아두시면 편리할 것 같습니다.

 

with open(‘화일명’,’모드’) as f:

   line = f.readlines()

   print(line)