파이썬 (python) - 외부파일 처리 ( CSV 파일 읽기, 쓰기 )

CSV(comma-separated values)는 정보를 컴마(,)로 구분한 텍스트 데이터 및 텍스트 파일이다. 확장자는 .csv입니다. 컴마

로 구분짓기에 .csv이고, 탭(tab)으로 구분짓는 파일은 .tsv입니다. 확장자를 .txt (텍스트파일)로 바꾸어도 열리며 읽을 수

있습니다.

1. CSV파일 읽기

일단 파이썬을 켜기전에 csv파일을 하나 만들어주겠습니다. 텍스트 파일에 컴마로 구분된 정보를 넣어주면 됩니다. 저는

커피가 좋으니까 커피 칼로리를 넣어보겠습니다.

다음 맨 뒤에 확장자만 csv로 바꾸어줍니다. 그리고 파일을 읽어서 출력해보면,

import csv
 
with open('./test_csv/coffee.csv','r') as f:
 
    readtest_csv = csv.reader(f)
 
    for c in readtest_csv:   #반복문으로 coffee.csv파일 내용 출력
        print(c)
 
================================================================================================================
 
>>PS C:\test_1> & C:/Users/hellg/AppData/Local/Programs/Python/Python38/python.exe c:/test_1/test_csv/test_1.py
['블랙커피 ', ' 0kcal']
['카푸치노 ', ' 100kcal']
['캔커피 ', ' 57kcal']
['카페오레 ', ' 57kcal']
['소주한잔 ', ' 90kcl']
 

이런식으로 반복문을 통해 출력해야합니다.

2. delimiter (구분문자) 옵션

delimiter은 말그대로 어떤 문자를 기준으로 구획을 나눌지 정하는 옵션입니다.

이번엔 컴마대신 @로 작성해서 출력해보겠습니다. 

import csv
 
with open('./test_csv/coffee.csv','r') as f:
    readtest_csv = csv.reader(f)
    for c in readtest_csv:   #반복문으로 coffee.csv파일 내용 출력
        print(c)
 
================================================================================================================
 
>>
['블랙커피 @ 0kcal']
['카푸치노 @ 100kcal']
['캔커피 @ 57kcal']
['카페오레 @ 57kcal']
['소주한잔 @ 90kcl']
 

컴마가 없으니까 한줄이 통째로 출력됩니다. 이때 delimiter을 사용하면 다음처럼 됩니다.

import csv
 
with open('./test_csv/coffee.csv','r') as f:
    readtest_csv = csv.reader(f, delimiter = '@')   #구획문자
    for c in readtest_csv:
        print(c)
 
 
================================================================================================================
 
>>
['블랙커피 ', ' 0kcal']
['카푸치노 ', ' 100kcal']
['캔커피 ', ' 57kcal']
['카페오레 ', ' 57kcal']
['소주한잔 ', ' 90kcl']
 

@를 기준으로 분할되어 출력되었습니다.

3. 딕셔너리로 출력하기 csv.DictReader(f)

4번줄 명령어의 csv.reader(f) 대신, csv.DictReader(f)를 사용하면 dictionary형태로 읽어옵니다.

4. writerow로 csv파일 쓰기.

import csv
 
w = [[11,1,2,3,4],['a','b','c','d','e'],[1,2,3,4,5]]
with open('./test_csv/coffee.csv','w', newline='') as f:   
# newline옵션은 줄바꿈을 없애줍니다. newline이 없으면 줄바꿈이 생깁니다.
    wt = csv.writer(f)
    for v in w:
        wt.writerow(v)
 
================================================================================================================
 
>>
 
 

덮어씌워지기 때문에 방금까지 존재했던 커피칼로리의 내용은 사라지고 변수 w에 저장된 문자열들이 삽입됩니다.

5. writerows 로 for문 없이 csv파일 바로쓰기

import csv
 
w = [[11,1,2,3,4],['a','b','c','d','e'],[1,2,3,4,5]]
with open('./csvwrite_1.csv','w', newline='') as f:
    wt = csv.writer(f)
    wt.writerows(w)  #for문 없이 바로쓰기.
 
================================================================================================================
 
>>
 
 

 4번과 결과는 같습니다.