[python] 파이썬 3 csv 파일 읽기

파이썬 버전 : 3.9.13

IDE : Spyder

파이썬으로 csv 파일을 읽어보겠습니다. test.csv 파일은 다음 내용이 들어가 있습니다.

id,name,rating

1,Akhil,4

2,Babu,3

3,Nikhil,5

헤더 라인 1줄과 데이터 라인 3줄로 이루어져 있습니다. 이 csv 파일을 읽어서 화면에 출력하는 샘플 코드입니다.

import csv

filename = "test.csv"

data = []

try:
    with open(filename) as f:
        reader = csv.reader(f)
        header = next(reader)
        data = [row for row in reader]    
except csv.Error as e:
    print("Error reading CSV file at line %s: %s" % (reader.line, e))
    exit(-1)
        
if header:
    print(header)
    print("================")
        
for datarow in data:
    print(datarow)

위 코드 실행 결과는 아래와 같습니다.

['id', 'name', 'rating']
================
['1', 'Akhil', '4']
['2', 'Babu', '3']
['3', 'Nikhil', '5']

끝.