[Python] csv 파일 쓰기, 읽기

csv 파일 쓰기

import csv

data = [
	["Regression", "qualification ", "rmse"],
    	["Classification", "quantification", "AUC"]
]

file = open("./models.csv", "w", newline="", encoding="utf-8-sig")
# newline : windows의 경우 한 줄씩 띄어지게 되는데 그것을 방지
# encoding="utf-8-sig" : windows의 경우, csv 파일이 vs code editor에서 csv 파일 깨지지 않게 설정
writer = csv.writer(file)
for d in data:
	writer.writerow(d)  # 한 줄씩 row로 쌓이게 됨
file.close()

 

csv 파일 읽기

import csv

file = open("./models.csv", "r")
reader = csv.reader(file)
for d in reader:
	print(d)
file.close()