Python

Python 언어를 사용하여 CSV 데이터 생성하기

지오준 2023. 11. 21.
반응형

CSV(Comma Separated Values) 파일은 데이터를 저장하고 전송하는 데에 널리 사용되는 형식입니다. 파이썬을 사용하여 CSV 파일을 생성하는 것은 매우 간단합니다. 이를 위해 csv 모듈을 사용할 수 있습니다.

import csv

# CSV 파일에 저장할 데이터
data = [
    ['이름', '나이', '직업'],
    ['제인', 25, '개발자'],
    ['스미스', 30, '디자이너'],
    ['앨리스', 27, '마케터']
]

# CSV 파일 생성 및 데이터 작성
file_name = 'data.csv'  # 파일 이름
with open(file_name, 'w', newline='') as file:
    writer = csv.writer(file)
    writer.writerows(data)

print(f'{file_name} 파일이 생성되었습니다.')

이 코드에서는 csv 모듈을 가져와 CSV 파일을 작성하기 위해 사용합니다. data 리스트에는 CSV 파일에 작성할 데이터가 포함되어 있습니다. 첫 번째 리스트는 열 제목으로, 나머지 리스트는 각 행의 데이터를 나타냅니다.

 

open() 함수를 사용하여 파일을 열고, csv.writer() 함수를 사용하여 CSV 파일을 작성합니다. writer.writerows() 함수를 통해 전체 데이터를 한 번에 작성할 수 있습니다.

 

위 코드를 실행하면, 같은 디렉토리에 data.csv 파일이 생성됩니다. 해당 파일을 열어보면 데이터가 쉼표로 구분되어 있는 것을 확인할 수 있습니다.

 

파이썬을 사용하여 CSV 파일을 생성하는 것은 이처럼 간단합니다. 여러분은 이를 통해 데이터를 저장하고 필요한 경우 다른 소프트웨어로 쉽게 전달할 수 있습니다. 파이썬의 csv 모듈은 CSV 파일을 다루는데 매우 유용한 도구입니다.

 

반응형

댓글