[Python] csv 파일 만들고, 안에 내용 쓰기

친구 과제 도와주다가 파이썬 공부하게 됨.

 

파이썬 공부하는 겸 정리하면 좋을 것 같아서 적게 됨.

import csv

def open_csv(file_name):
    file_name += '.csv'
    with open(file_name, 'r') as f:
        data_list = csv.reader(f)
        return list(data_list)

def write_csv(file_name, data_list):
    file_name += '.csv'
    with open(file_name, 'w', newline='') as f:
        csv_obj = csv.writer(f, delimiter=',')
        csv_obj.writerows(data_list)

list_info = [['구', '전체', '내국인', '외국인'],
             ['관악구', 200, 150, 50],
             ['강남구', 400, 310, 90],
             ['송파구', 380, 300, 80],
             ['강동구', 420, 310, 110]]

write_csv('함수만들기', list_info)
print(open_csv('함수만들기'))