여정의 기록

캐글에서 .csv 파일 로컬에 다운로드하고 싶을 때 본문

공부/프로젝트 진행

캐글에서 .csv 파일 로컬에 다운로드하고 싶을 때

Chelsey 2022. 10. 2. 18:20
728x90

어떻게 만들었는지는 몰라도 누군가가 만들어둔 데이터를 보면 다운로드 받고싶을 때가 있습니다.

test_df 라는 이미 불러온 csv파일을 저장하고 싶었습니다.
맨 마지막 줄에 보면 create_download_link 함수의 인자로 test_df를 넣은것을 볼 수 있습니다.

import base64
import pandas as pd
from IPython.display import HTML

def create_download_link(df, title = "Download CSV file", filename = "test_df.csv"):
    csv = df.to_csv()
    b64 = base64.b64encode(csv.encode())
    payload = b64.decode()
    html = '<a download="{filename}" href="data:text/csv;base64,{payload}" target="_blank">{title}</a>'
    html = html.format(payload=payload,title=title,filename=filename)
    return HTML(html)

create_download_link(test_df)

 

 

출처 : https://stackoverflow.com/questions/31893930/download-csv-from-an-ipython-notebook

728x90