[Python] CSV 파일 읽기/쓰기

 

1. CSV 파일 읽기

먼저 아래와 같은 CSV 파일을 만들어놓았다

 

 

이를 읽어서 콘솔에 출력하기 위해 아래와 같은 파이썬 파일을 만들었다

import csv

path = 'C:/sample.csv'

try:
	f = open(path, encoding='euc-kr')
	csv_f = csv.reader(f)

	for line in csv_f:
		print(line[0], '\t', line[1], '\t', line[2], '\t', line[3], '\t', line[4])

except Exception as e:
    print(e)

csv를 import 하면, reader 함수로 지정한 경로의 CSV 파일을 읽을 수 있다

try/except 구문은 예외처리를 위한 것으로, 아래 글을 참조하자

wayhome25.github.io/python/2017/02/26/py-12-exception/

 

또한, 한글 데이터가 들어있기 때문에 euc-kr로 인코딩해주었다

 

cmd에서 실행해보면

 

 

해당 파일의 데이터들이 잘 나온다

 

 

2. CSV 파일 쓰기

쓰기는 reader 대신 writer를 쓰면 된다

다만 권한을 꼭 지정해줘야 하는데, 단순 쓰기라 생각해서 'w'라 넣으면 그냥 덮어쓰기가 되므로 행 추가를 위해선 append 의미의 'a'를 추가해주자

import csv

path = 'C:/sample.csv'

try:
	f = open(path, 'a', encoding='euc-kr', newline='')
	csv_f = csv.writer(f)

	csv_f.writerow([400, '인턴', '이', '2019-04-01', 24])

except Exception as e:
    print(e)

실행해보면

 

 

밑에 잘 추가되어 나온다

 

깃헙에서 코드 확인