[Python] 파이썬 CSV 모듈 쓰기, 읽기, 수정

파이썬 CSV 모듈 쓰기, 읽기, 수정

 

안녕하세요 :-)

파이썬에서 csv 파일에 쓰고, 읽고, 수정(추가)하는 방법입니다.

파이썬 언어로는 주로 데이터를 다룰 텐데 도출된 데이터 혹은 중간 데이터를 엑셀 또는 csv 파일로 내보내는 것이 필요할 때가 있습니다.
파이썬에는 csv 모듈이 있어서 간단하게 구현이 가능합니다.

참고로 CSV란 Comma Separated Values로 '쉼표로 구분된 값'이라는 뜻입니다. 

 

1. 파일 읽기

import csv
with open('filename.csv', 'r', newline='', encoding='utf-8-sig') as csv_file:
    reader = csv.reader(csv_file, delimiter=',', quotechar='"')
    for row in reader:
        print(", ".join(row))

#1 csv 모듈을 import 합니다.
#2 csv_file 이름으로 'filename.csv' 파일을 'r' 옵션으로 읽습니다.
#3 reader라는 객체로 한줄 한 줄 읽어옵니다.
#4 각 행은 문자열 리스트로 반환됩니다.

 

2. 파일 쓰기

import csv
header = ['first', 'second', 'third']
with open('filename.csv', 'w', newline='') as csv_file:
    writer = csv.writer(csv_file, delimiter=',', quotechar='"')
    writer.writerow(header)
    writer.writerow(['first value', 'second value', 'third value'])

#1 csv 모듈을 import 합니다.
#2 header를 정의합니다.
#3 csv_file 이름으로 'filename.csv' 파일을 'w' 옵션으로 씁니다.
#4 writer라는 객체를 생성합니다.
#5 writerow를 통해 헤더를 쓰고, 아래줄에 넣을 값을 입력합니다.

 

3. 파일 수정

파일 수정은 간단합니다. 'r', 'w' 옵션 자리에 'a'를 넣으면 append의 개념으로 아래 줄에 추가가 됩니다.

 

그럼 궁금한게 있다면 아래 댓글로 남겨주세요!

이만 :)