캐글을 하면 output을 다운받고 싶지만 커밋하기에는 시간이 너무 많이 걸리는 상황이 있습니다.

이때 꼭 Commit을 해야만 Output 파일을 만들 수 있을까요?? NO!

다음과 같이 다운로드 링크를 만드는 방법도 있습니다.

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

def create_download_link( df, title = "Download CSV file", filename = "data.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)

df = pd.DataFrame(data = [[1,2],[3,4]], columns=['Col 1', 'Col 2'])
create_download_link(df)

그럼 다들 즐거운 Kaggle 생활!!

Reference

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

Leave a Comment