[파이썬] CSV 파일의 읽기와 쓰기

CSV (Comma Separated Values) 파일은 데이터를 텍스트 형식으로 저장하는 가장 일반적인 방법 중 하나입니다. 파이썬에서는 CSV 파일을 쉽게 읽고 쓸 수 있는 기능을 제공합니다. 이번 포스트에서는 파이썬에서 CSV 파일을 읽고 쓰는 방법에 대해 알아보겠습니다.

CSV 파일 읽기

import csv

# CSV 파일을 읽기 모드로 열기
with open('data.csv', 'r') as file:
    # CSV 파일을 읽는 CSV Reader 객체 생성
    reader = csv.reader(file)

    # 각 행을 순회하며 데이터 출력
    for row in reader:
        print(row)

위의 예제 코드에서는 csv 모듈을 사용하여 CSV 파일을 읽습니다. with 문을 사용하여 파일을 열고, csv.reader를 사용하여 파일을 읽는 reader 객체를 생성합니다. 그리고 for 문을 사용하여 각 행을 순회하며 데이터를 출력합니다.

CSV 파일 쓰기

import csv

# CSV 파일을 쓰기 모드로 열기
with open('data.csv', 'w', newline='') as file:
    # CSV 파일을 쓰는 CSV Writer 객체 생성
    writer = csv.writer(file)

    # 데이터 쓰기
    writer.writerow(['Name', 'Age', 'Location'])
    writer.writerow(['John Doe', '30', 'New York'])
    writer.writerow(['Jane Smith', '25', 'London'])

위의 예제 코드에서는 csv 모듈을 사용하여 CSV 파일을 씁니다. with 문을 사용하여 파일을 열고, csv.writer를 사용하여 파일을 쓰는 writer 객체를 생성합니다. 그리고 writerow 메소드를 호출하여 데이터를 씁니다. 각 writerow 호출은 파일에 새로운 행을 추가합니다.

추가적인 옵션

위의 예제 코드에서는 기본적인 CSV 파일의 읽기와 쓰기를 다루었습니다. 그러나 csv 모듈은 다양한 옵션을 제공하여 CSV 파일을 보다 유연하게 다룰 수 있습니다. 예를 들어, 다른 구분자를 사용하거나 특정 열만을 선택하여 읽거나 쓸 수도 있습니다. 자세한 사용 방법과 옵션에 대해서는 Python 공식 문서를 참고하세요.

CSV 파일을 다루는 것은 데이터 처리와 분석 작업에서 매우 중요합니다. 파이썬의 csv 모듈을 활용하면 CSV 파일을 쉽게 읽고 쓸 수 있으므로, 데이터 작업에 이 모듈을 적극적으로 활용해 보세요.