Python 리스트(list) csv 파일로 읽고 쓰기(불러오기, 저장하기)

딥러닝 등 AI관련 작업을 하다보면 csv를 가져와서 열고 작업을 하는 경우가 제법 많다. 그런데, 막상 직접 만든 코드에서csv를 뽑아내고 그 파일을 불러들이려다 보면 코드 작성이 헷갈릴 때가 있따.

그래서 특정 예제를 만들어 csv파일을 읽는 과정과 쓰는 과정을 각각 보이도록 하겠다.

Python 리스트 csv로 저장하기

파이썬 모듈인 csv 모듈을 import하고 데이터는 날짜 데이터, 정수 데이터, 부동소수점 자료형 데이터가 있다고 가정했다. 엑셀로 그린다면 열 방향으로 그릴 것 같지만, 파이썬 리스트로 표현하여 행 방향으로 나열되었다.

# csv 모듈 import하기
import csv

# list 만들기
date = ['200801','200802','200803','200804','200805']
val1 = [50,100,300,150,200]
val2 = [2000.0, 2200.5, 2320.3, 2250.0, 2200.0]

# csv파일로 적기
# newline 설정을 안하면 한줄마다 공백있는 줄이 생긴다.
with open('listfile.csv', 'w', newline='') as f:
    writer = csv.writer(f)
    writer.writerow(date)
    writer.writerow(val1)
    writer.writerow(val2)

csv파일 출력결과:

>>
200801,200802,200803,200804,200805
50,100,300,150,200
2000.0,2200.5,2320.3,2250.0,2200.0

Python에서 저장했던 csv파일 리스트로 불러오기

앞서 행방향으로 나열했던 각 줄(line)을 csv 모듈을 이용하면 바로 리스트로 불러들이는 것이 가능하다.

# csv 파일 읽기
with open('listfile.csv', 'r', encoding='utf-8') as f:
    rdr = csv.reader(f)
    for i,line in enumerate(rdr):
        if i==0:
            date = line
        elif i==1:
            val1 = line
        elif i==2:
            val2 = line

이 경우에는 각 라인이 어떤 데이터를 아는 경우이기 때문에 줄마다 특정 리스트 자료형을 만들어 리스트를 받아왔다. 일반적으로 파일 불러오기에서는 readline해서 불러온 라인이 str자료형인데, 이 경우에는 바로 list 자료형으로 들어오는 것을 확인할 수 있다.

확인하기 위해 아래와 같이 코드를 작성해서 결과를 보자.

print(date)
print(val1)
print(val2)

출력결과:

>>
['200801', '200802', '200803', '200804', '200805']
['50', '100', '300', '150', '200']
'2000.0', '2200.5', '2320.3', '2250.0', '2200.0']