파이썬을 이용한 CSV 저장
서론
CSV(Comma-Separated Values)는 데이터를 쉼표(,)로 구분하여 저장하는 파일 형식입니다. 파이썬에서는 csv 모듈을 사용하여 CSV 파일을 읽고 쓸 수 있습니다. 이 포스트에서는 파이썬을 이용하여 CSV 파일을 저장하는 방법에 대해 알아보겠습니다.
본론
- csv 모듈 import하기: 먼저, csv 모듈을 import하여 사용할 준비를 합니다.
python
import csv
- CSV 파일 쓰기: CSV 파일을 쓰기 위해서는
open()함수를 사용하여 파일을 열고,csv.writer()를 이용하여 writer 객체를 생성해야 합니다. 그리고writerow()메서드를 사용하여 데이터를 한 줄씩 파일에 쓸 수 있습니다.
python
with open('data.csv', 'w', newline='') as file:
writer = csv.writer(file)
writer.writerow(['이름', '나이'])
writer.writerow(['John', '25'])
writer.writerow(['Amy', '30'])
위의 코드는 ‘data.csv’라는 파일을 쓰기 모드로 열고, writer 객체를 생성한 뒤, 각각의 데이터를 파일에 씁니다. 첫 번째 writerow() 호출은 헤더를 쓰고, 두 번째와 세 번째 writerow() 호출은 실제 데이터를 씁니다. 데이터는 리스트 형식으로 제공되어야 합니다.
- CSV 파일에 딕셔너리 쓰기: 만약 데이터를 딕셔너리 형식으로 제공하고 싶다면,
csv.DictWriter()를 사용하여 딕셔너리 값을 CSV 파일에 쓸 수 있습니다. 다음의 예제 코드를 참고하세요.
“`python
with open(‘data.csv’, ‘w’, newline=”) as file:
fieldnames = [‘이름’, ‘나이’]
writer = csv.DictWriter(file, fieldnames=fieldnames)
writer.writeheader()
writer.writerow({'이름': 'John', '나이': '25'})
writer.writerow({'이름': 'Amy', '나이': '30'})
“`
위의 코드는 DictWriter()를 사용하여 딕셔너리 형식의 데이터를 CSV 파일에 쓰는 방법을 보여줍니다. 딕셔너리의 키 값은 CSV 파일의 헤더로 사용됩니다.
결론
이 포스트에서는 파이썬을 이용하여 CSV 파일을 저장하는 방법에 대해 알아보았습니다. csv 모듈을 사용하여 CSV 파일을 열고 쓸 수 있으며, 데이터를 리스트 형식이나 딕셔너리 형식으로 제공할 수 있습니다. CSV 파일을 쓰는 방법에 대한 예제 코드를 통해 실제 데이터를 저장하는 방법을 이해하였습니다.