파이썬에서 CSV 파일 읽기 및 쓰기

CSV(Comma-Separated Values) 파일은 데이터 저장 및 교환에 널리 사용됩니다. Python에서는 내장된 csv 모듈을 사용하여 CSV 파일을 쉽게 읽고 쓸 수 있습니다. 이 문서에서는 Python에서 CSV 파일을 사용하는 기본 사항을 안내합니다.

CSV 파일 읽기

CSV 파일을 읽으려면 csv 모듈의 csv.reader 클래스를 사용할 수 있습니다. 간단한 예는 다음과 같습니다.

import csv

# Open the CSV file
with open('example.csv', mode='r') as file:
    reader = csv.reader(file)
    
    # Iterate over each row in the CSV
    for row in reader:
        print(row)

이 예에서 example.csv은 읽기 모드로 열리고, csv.reader은 파일을 구문 분석하는 데 사용됩니다. CSV 파일의 각 행은 값 목록으로 인쇄됩니다.

CSV 파일 쓰기

CSV 파일에 데이터를 쓰려면 csv.writer 클래스를 사용합니다. 다음은 CSV 파일에 행 목록을 쓰는 방법입니다.

import csv

# Data to be written to the CSV
data = [
    ['Name', 'Age', 'City'],
    ['Alice', 30, 'New York'],
    ['Bob', 25, 'Los Angeles'],
    ['Charlie', 35, 'Chicago']
]

# Open the CSV file in write mode
with open('output.csv', mode='w', newline='') as file:
    writer = csv.writer(file)
    
    # Write each row to the CSV file
    writer.writerows(data)

이 예에서 output.csv은 지정된 데이터로 생성되거나 덮어씁니다. writer.writerows() 메서드는 여러 행을 한 번에 쓰는 데 사용됩니다.

결론

파이썬에서 CSV 파일을 다루는 것은 csv 모듈을 사용하면 간단합니다. CSV 파일을 쉽게 읽고 쓸 수 있어 이 일반적인 형식의 데이터를 처리하기 쉽습니다.