본문으로 건너뛰기

파이썬에서 CSV 파일에 쓰는 방법

파이썬에서 CSV 파일에 쓰는 방법에 대한 상세한 단계별 튜토리얼입니다.

단계 1: CSV 모듈 가져오기​

먼저 CSV 파일을 작업하기 위한 기능을 제공하는 csv 모듈을 가져와야 합니다. 다음 코드를 사용하여 모듈을 가져올 수 있습니다.

import csv

단계 2: 쓰기 모드로 CSV 파일 열기​

다음으로, CSV 파일을 쓰기 모드로 열어야 합니다. 이렇게 하면 파일이 존재하지 않으면 새 파일이 생성되거나 기존 파일이 덮어씌워집니다. open() 함수를 사용하여 파일을 열고, 파일 이름과 모드로 'w'를 지정합니다.

with open('file.csv', 'w', newline='') as csvfile:
    # 파일에 쓰기 위한 코드

newline='' 인수는 CSV 파일의 행 사이에 추가적인 공백 줄이 삽입되는 것을 방지하기 위해 필요합니다.

단계 3: CSV Writer 객체 생성하기​

파일을 열었다면, CSV Writer 객체를 생성해야 합니다. 이 객체를 사용하여 데이터를 CSV 파일에 쓸 수 있습니다. csv.writer() 함수를 사용하여 파일 객체를 매개변수로 전달하여 Writer 객체를 생성할 수 있습니다.

with open('file.csv', 'w', newline='') as csvfile:
    writer = csv.writer(csvfile)
    # 파일에 쓰기 위한 코드

단계 4: CSV 파일에 행 쓰기​

이제 CSV Writer 객체가 준비되었으므로, CSV 파일에 행을 쓸 수 있습니다. 각 데이터 행은 열을 나타내는 요소로 이루어진 리스트여야 합니다. 행을 쓰기 위해 CSV Writer 객체의 writerow() 메서드를 사용할 수 있습니다.

with open('file.csv', 'w', newline='') as csvfile:
    writer = csv.writer(csvfile)
    writer.writerow(['이름', '나이', '국가'])  # 헤더 행 쓰기
    writer.writerow(['존', 25, '미국'])         # 데이터 행 쓰기
    writer.writerow(['앨리스', 32, '캐나다'])

위의 예시에서는 먼저 열 이름을 가진 헤더 행을 쓰고, 그 다음에 데이터 행 두 개를 씁니다.

단계 5: CSV 파일 닫기​

CSV 파일에 쓰기를 마치면, 파일을 제대로 닫는 것이 중요합니다. 이렇게 하면 모든 데이터가 기록되고 파일과 관련된 시스템 리소스가 해제됩니다. 파일 객체의 close() 메서드를 사용하여 파일을 닫을 수 있습니다.

with open('file.csv', 'w', newline='') as csvfile:
    writer = csv.writer(csvfile)
    # 파일에 쓰기 위한 코드

# 'with' 블록 외부에서 파일이 자동으로 닫힙니다.

with 문을 사용하면, 코드 블록을 종료할 때 파일이 자동으로 닫힙니다. 예외가 발생하더라도 파일은 자동으로 닫힙니다.

여기까지입니다! 이제 파이썬에서 CSV 파일에 쓰는 방법을 알았습니다. 특정 요구사항과 데이터에 맞게 코드를 사용자 정의하도록 기억하세요.