[파이썬 파일처리] 파이썬 CSV 파일 읽기, 한 줄씩 데이터 가져오는 방법

파이썬 CSV 파일 읽기는 저장보다 더 자주 쓰는 작업이다.

이미 있는 데이터를 불러와서 출력하거나 가공할 때 기본이 된다.

엑셀 파일보다 가볍고 다루기 쉬워서 CSV를 먼저 익혀두면 실무에 도움이 된다.

 

- 가장 기본적인 CSV 읽기

아래 코드는 CSV 파일 내용을 한 줄씩 읽어서 출력하는 예제다.

import csv

with open("people.csv", "r", newline="", encoding="utf-8-sig") as file:
    reader = csv.reader(file)
    for row in reader:
        print(row)

 

- 실행 결과

위 코드를 실행하면 각 행이 리스트 형태로 출력된다.

['이름', '나이', '지역']
['민수', '27', '서울']
['지은', '31', '부산']
['현우', '29', '대전']

 

- 특정 열만 가져오려면

이름만 출력하고 싶다면 인덱스로 접근하면 된다.

import csv

with open("people.csv", "r", newline="", encoding="utf-8-sig") as file:
    reader = csv.reader(file)
    next(reader)

    for row in reader:
        print(row[0])

 

- 코드 설명

• csv.reader()는 CSV 내용을 한 줄씩 읽는다.
• next(reader)는 첫 줄 제목 행을 건너뛴다.
• row[0]은 첫 번째 열을 의미한다.

 

- 자주 하는 실수

1. 한글 CSV를 읽는데 encoding을 안 맞춰서 깨지는 경우
2. 제목 행까지 데이터로 처리하는 경우
3. row[3]처럼 존재하지 않는 열 번호를 써서 오류가 나는 경우

 

- 정리하며

파이썬 CSV 파일 읽기는 csv.reader만 익혀도 바로 시작할 수 있다.

제목 행 건너뛰기와 열 인덱스 접근까지 익혀두면 대부분의 기본 작업은 처리할 수 있다.