[Python/파이썬] CSV 파일 읽기, 쓰기

CSV(Comma-Separated Values) 파일은 데이터를 저장하는 데 널리 사용되는 형식입니다. 파이썬에서는 CSV 파일을 읽고 쓰는 데 사용할 수 있는 여러 가지 방법이 있습니다. 아래 예제를 통해 각 방법을 살펴보겠습니다.

1. CSV 파일 읽기

CSV 파일을 읽어서 데이터를 파이썬 코드로 가져올 수 있습니다. csv 모듈을 사용하여 파일을 열고 데이터를 읽습니다.

import csv

with open('data.csv', 'r', newline='') as file:
    reader = csv.reader(file)
    for row in reader:
        print(row)

위 코드에서는 csv.reader() 함수를 사용하여 파일을 읽고 각 행을 반복하여 출력합니다.

2. CSV 파일 쓰기

데이터를 CSV 파일로 쓰려면 csv 모듈을 사용하여 파일을 열고 데이터를 씁니다.

import csv

data = [
    ['Name', 'Age', 'Country'],
    ['John', 30, 'USA'],
    ['Emily', 28, 'Canada'],
    ['Tom', 35, 'UK']
]

with open('output.csv', 'w', newline='') as file:
    writer = csv.writer(file)
    writer.writerows(data)

위 코드에서는 csv.writer() 함수를 사용하여 데이터를 파일에 씁니다. writerows() 메서드를 사용하여 모든 행을 한 번에 씁니다.