[Python/파이썬] 딕셔너리 CSV 파일 쓰기

CSV(Comma Separated Values) 파일은 데이터를 텍스트 파일 형식으로 저장하는 데 사용되며, 각 열은 쉼표(,)로 구분됩니다. Python에서는 CSV 파일을 쓰고 읽는 데 유용한 모듈인 csv를 제공합니다.

1. 딕셔너리를 CSV 파일로 쓰기

딕셔너리를 CSV 파일로 쓰려면 csv 모듈의 DictWriter 클래스를 사용합니다. DictWriter 클래스를 초기화할 때 필드 이름과 출력 파일을 지정합니다.

import csv

data = [
    {'Name': 'John', 'Age': 30, 'City': 'New York'},
    {'Name': 'Emma', 'Age': 35, 'City': 'Los Angeles'}
]

fields = ['Name', 'Age', 'City']

with open('data.csv', 'w', newline='') as csvfile:
    writer = csv.DictWriter(csvfile, fieldnames=fields)
    writer.writeheader()
    writer.writerows(data)

2. CSV 파일에서 딕셔너리 읽기

CSV 파일에서 딕셔너리를 읽으려면 csv 모듈의 DictReader 클래스를 사용합니다. DictReader 클래스를 초기화할 때 입력 파일을 지정합니다.

import csv

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