在Watson Studio中下载Pandas Dataframe

Question

我想知道当我在Watson Studio中使用Jupyter Notebok时如何以CSV文件形式下载Pandas Dataframe。

Answer 1

我假设您已经创建了一个pandas数据帧，现在想知道在哪里可以将数据帧保存为csv文件，然后最终将保存的csv文件下载到本地计算机。

您需要使用项目api将pandas数据框保存为csv到项目数据资产，如以下链接所述： -

# Save dataframe as csv file to storage
project.save_data(data=df.to_csv(index=False),file_name='iris1.csv',overwrite=True)

https://medium.com/ibm-data-science-experience/control-your-dsx-projects-using-python-c69e13880312

将其保存为数据资产后，您可以在项目中使用此数据资产旁边的3点菜单将该数据资产下载到本地计算机。

我希望它有所帮助。

Answer 2

另一种方法是编写一个函数，该函数首先对数据进行base64编码，然后将其嵌入到下载链接中：

# Download as CSV: data frame, optional title and filename
def create_download_link_csv(df, title = "Download CSV file", filename = "data.csv"):  
    # generate in-memory CSV, then base64-encode it
    csv = df.to_csv(index=False)
    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)

然后，您将使用DataFrame作为第一个参数调用该函数，它将返回下载链接。相似但稍微复杂的东西也适用于Excel文件。我有一些background information on CSV and Excel download as well as a complete Gist in my blog entry。