Python으로 CSV 읽기

파이썬에서 CSV 파일을 읽고 쓰는 방법에 대해 알아보겠습니다. CSV 파일은 Comma-Separated Values의 약자로, 쉼표로 구분된 텍스트 데이터를 저장하는 파일 형식입니다. CSV 파일은 엑셀이나 구글 스프레드시트와 같은 스프레드시트 프로그램에서 쉽게 열고 편집할 수 있으며, 다양한 데이터 분석이나 머신러닝 작업에도 유용하게 사용됩니다.

 

파이썬에서 CSV 파일을 다루기 위해서는 내장 모듈인 csv를 임포트해야 합니다.

csv 모듈은 CSV 파일을 읽고 쓰는 데 필요한 여러 가지 함수와 클래스를 제공합니다. 예를 들어, csv.reader() 함수는 CSV 파일을 열고 각 행을 리스트로 변환하는 반복자(iterator)를 반환합니다.

csv.writer() 함수는 CSV 파일을 쓰기 모드로 열고 각 행을 리스트로 받아서 쉼표로 구분된 문자열로 변환하는 객체를 반환합니다.

CSV 파일을 읽는 가장 간단한 방법은 다음과 같습니다.

python
import csv

# CSV 파일 열기
with open('example.csv', 'r') as f:
# csv.reader 객체 생성
reader = csv.reader(f)
# 각 행을 리스트로 읽기
for row in reader:
print(row)

이 코드는 example.csv라는 파일을 읽기 모드로 열고, csv.reader 객체를 생성하여 각 행을 리스트로 읽어서 출력합니다. 예를 들어, example.csv 파일의 내용이 다음과 같다면,

name,age,gender
Alice,20,F
Bob,25,M
Charlie,30,M

코드의 출력 결과는 다음과 같습니다.

['name', 'age', 'gender']
['Alice', '20', 'F']
['Bob', '25', 'M']
['Charlie', '30', 'M']

CSV 파일을 쓰는 가장 간단한 방법은 다음과 같습니다.

import csv

# CSV 파일 열기
with open('output.csv', 'w') as f:
# csv.writer 객체 생성
writer = csv.writer(f)
# 헤더 행 쓰기
writer.writerow(['name', 'age', 'gender'])
# 데이터 행 쓰기
writer.writerow(['David', '35', 'M'])
writer.writerow(['Eve', '40', 'F'])
writer.writerow(['Frank', '45', 'M'])

이 코드는 output.csv라는 파일을 쓰기 모드로 열고, csv.writer 객체를 생성하여 헤더 행과 데이터 행을 리스트로 받아서 쉼표로 구분된 문자열로 변환하여 씁니다. 예를 들어, 코드의 실행 결과로 생성된 output.csv 파일의 내용은 다음과 같습니다.

name,age,gender
David,35,M
Eve,40,F
Frank,45,M

csv 모듈은 CSV 파일을 다루는 데 더 많은 기능과 옵션을 제공합니다. 예를 들어, csv.DictReader() 클래스는 CSV 파일의 각 행을 딕셔너리로 변환하는 반복자를 반환하며, csv.DictWriter() 클래스는 딕셔너리를 받아서 CSV 파일에 쓰는 객체를 반환합니다. 또한, delimiter 파라미터를 사용하면 쉼표 대신 다른 문자를 구분자로 사용할 수 있습니다. 자세한 내용은 공식 문서(https://docs.python.org/3/library/csv.html)를 참고하세요.