파이썬 CSV 파일 읽기/쓰기: 종합 가이드

파이썬 CSV 파일 읽기/쓰기: 종합 가이드

CSV 파일이란?

CSV (Comma-Separated Values) 파일은 표 형식 데이터를 일반 텍스트로 저장합니다. 각 줄은 한 행을 나타내며, 개별 필드는 쉼표, 세미콜론 또는 다른 구분자로 분리됩니다. CSV 파일은 가볍고 스프레드시트, 데이터베이스, 데이터 분석 도구에서 폭넓게 지원되어 데이터 교환용 선호 포맷입니다.

CSV 샘플 파일

아래는 작은 프로그래밍 언어 데이터베이스의 전형적 CSV 표현. 첫 행은 열 헤더, 이후 데이터 행들.

표 데이터

CSV 데이터

Programming language,Designed by,Appeared,Extension
Python,Guido van Rossum,1991,.py
Java,James Gosling,1995,.java
C++,Bjarne Stroustrup,1983,.cpp

파이썬 내장 csv 모듈

csv 모듈은 CSV 파일 읽기/쓰기 저수준 도구 제공. 인용, 구분자, 줄 종료자 자동 처리해 수동 문자열 조작 필요성 감소.

주요 함수

• csv.reader() – 행을 리스트로 순회.
• csv.DictReader() – 열 이름으로 키된 딕셔너리로 행 변환.
• csv.writer() – 행을 CSV 파일에 쓰기.
• csv.DictWriter() – 딕셔너리를 CSV 파일에 쓰기.
• 방언 헬퍼: csv.get_dialect(), csv.list_dialects(), csv.register_dialect(), csv.unregister_dialect().

CSV 파일 읽기

CSV 파일 읽고 각 행을 파이썬 리스트로 출력하는 최소한의 프로덕션 레디 예제:

import csv
with open('data.csv', mode='r', newline='', encoding='utf-8') as file:
    reader = csv.reader(file)
    for row in reader:
        print(row)

출력:

["Programming language", "Designed by", "Appeared", "Extension"]
["Python", "Guido van Rossum", "1991", ".py"]
["Java", "James Gosling", "1995", ".java"]
["C++", "Bjarne Stroustrup", "1983", ".cpp"]

CSV를 딕셔너리로 읽기

csv.DictReader 사용 시 헤더 행이 자동으로 딕셔너리 키로 매핑되어 데이터 처리 간소화.

import csv
with open('data.csv', mode='r', newline='', encoding='utf-8') as file:
    reader = csv.DictReader(file)
    for row in reader:
        print(row)

결과 딕셔너리:

{"Programming language": "Python", "Designed by": "Guido van Rossum", "Appeared": "1991", "Extension": ".py"}
{"Programming language": "Java", "Designed by": "James Gosling", "Appeared": "1995", "Extension": ".java"}
{"Programming language": "C++", "Designed by": "Bjarne Stroustrup", "Appeared": "1983", "Extension": ".cpp"}

CSV 파일 쓰기

csv.writer와 csv.DictWriter 클래스로 효율적 데이터 쓰기.

import csv
rows = [
    ["Programming language", "Designed by", "Appeared", "Extension"],
    ["Python", "Guido van Rossum", "1991", ".py"],
    ["Java", "James Gosling", "1995", ".java"],
    ["C++", "Bjarne Stroustrup", "1983", ".cpp"],
]
with open('writeData.csv', mode='w', newline='', encoding='utf-8') as file:
    writer = csv.writer(file, delimiter=',', quotechar='"', quoting=csv.QUOTE_MINIMAL)
    writer.writerows(rows)

생성된 writeData.csv는 원본 샘플과 일치.

Pandas로 CSV 작업

Pandas는 데이터를 DataFrame에 로드해 최소 코드로 강력한 데이터 조작 가능. 미설치 시 pip install pandas로 설치.

Pandas로 읽기

import pandas as pd
df = pd.read_csv('data.csv')
print(df)

출력:

  Programming language Designed by Appeared Extension
0 Python Guido van Rossum 1991 .py
1 Java James Gosling 1995 .java
2 C++ Bjarne Stroustrup 1983 .cpp

Pandas로 쓰기

import pandas as pd
data = {
    'Programming language': ['Python', 'Java', 'C++'],
    'Designed by': ['Guido van Rossum', 'James Gosling', 'Bjarne Stroustrup'],
    'Appeared': ['1991', '1995', '1983'],
    'Extension': ['.py', '.java', '.cpp'],
}
df = pd.DataFrame(data)
df.to_csv('pandaresult.csv', index=False)
print(df)

결과 CSV는 원본과 동일하지만 DataFrame은 고급 분석 기능 제공.

결론

파이썬 csv 모듈과 Pandas는 CSV 파일 읽기, 쓰기, 조작을 위한 견고하고 사용하기 쉬운 도구를 제공합니다. 이 라이브러리들을 마스터하면 웹 서비스, ETL 파이프라인, 과학 연구 전반에서 효율적 데이터 워크플로 구축 가능. 오늘부터 실험해보고 CSV 처리를 프로젝트에 자신 있게 통합하세요.