Python | CSV 파일과 CSV파일 읽고, 쓰기

728x90

csv 파일이 뭔지, 이를 파이썬으로 어떻게 읽고, 쓰는지 알아보자.

 

 

In [5]:

# CSV 는 Comma Seperated Value 의 약자로 쉼표로 구분된 파일

# 데이터들을 쉼표로 구분하는 것이 특징이다.

# csv 데이터 형식의 데이터 생성
billboardchart = {
                 1 : ["Tho Box","Roddy Ricch","2019-12-19"],
                 2 : ["Don't Start Now", "Dua Lipa", "2019-11-01"],
                 3 : ["Life Is Good", "Future Featuring Drake", "2020-02-10"],
                 4 : ["Blinding", "The Weeknd", "2019-11-29"],
                 5 : ["Circles", "Post Malone","2019-08-30"]}

with open("billboardchart.csv","w") as f:
    
    # # 위에서 생성한 csv 데이터를 i에 저장
    for i in billboardchart.values():
        
        # 각 데이터를 쉼표로 구분
        data = ",".join(i)
        
        # 데이터 쓰기모드, 각 데이터 별 줄 바꾸기
        f.write(data+"\n")
        
# csv 파일을 읽어오기위한 csv 패키지 가져오기
import csv

# 추가해주기 위한 카테고리명을 리스트로 저장
header = ["title", "singer", "released date"]

# csv 파일을 읽어오기 모드로 열기
with open("billboardchart.csv","r") as inputfile:
    
    # csv 파일을 쓰기모드로 열기
    with open("billboardchart_out.csv","w", newline='\n') as outputfile:
        
        # fi 변수에 위에서 저장한 csv 파일을 가져와 저장
        fi = csv.reader(inputfile, delimiter=',')
        
        # fo 변수에 
        fo = csv.writer(outputfile, delimiter=',')
        fo.writerow(header)
        for row in fi:
            fo.writerow(row)    
            
# 컬렁으로 사용할 fields 값과 데이터로 사용할 rows 값 저장
fields = ["title", "singer", "released date"]
rows = [ ["Tho Box","Roddy Ricch","2019-12-19"],
               ["Don't Start Now", "Dua Lipa", "2019-11-01"],
               ["Life Is Good", "Future Featuring Drake", "2020-02-10"],
               ["Blinding", "The Weeknd", "2019-11-29"],
               ["Circles", "Post Malone","2019-08-30"]]
            
# 판다스를 이용해 데이터를 csv 파일로 저장
# 판다스 패키지 가져오기
import pandas as pd

# 판다스의 데이터프레임을 통해 파일열기
# pd.DataFrame(가져올 데이터, columns = 컬렁명) 의 서식으로 사용
df = pd.DataFrame(rows, columns = fields)

# pandas.csv 로 저장
df.to_csv('pandas.csv',index = False)


# csv.writer를 통한 수정
import csv 

# 파일명 저장
filename = "test.csv"

# csv 파일을 열기 및 반복문을 통해 csv_writer 변수에 데이터 저장
with open(filename, 'w+', newline='\n') as csv_file: 
    csv_writer = csv.writer(csv_file) 
    csv_writer.writerow(fields) 
    csv_writer.writerows(rows)

df = pd.read_csv('pandas.csv')
df.head()

Out[5]:

	title	singer	released date
0	Tho Box	Roddy Ricch	2019-12-19
1	Don't Start Now	Dua Lipa	2019-11-01
2	Life Is Good	Future Featuring Drake	2020-02-10
3	Blinding	The Weeknd	2019-11-29
4	Circles	Post Malone	2019-08-30