python - pd.ExcelWriter, writer.save(), 下载文件到本地

Question

我在 databricks 环境中使用 Python 3.7.5。
我编写了一个非常简单的函数，将多个数据帧连接到一个 Excel 电子表格中。 主要问题是我无法访问要下载到本地计算机的文件。

# function

def dfs_tabs(df_list, sheet_list, file_name):

    writer = pd.ExcelWriter(file_name,engine='xlsxwriter')  

    for dataframe, sheet in zip(df_list, sheet_list):

        dataframe.to_excel(writer, sheet_name=sheet, startrow=0 , startcol=0)  

    writer.save()

 

# list of dataframes and sheet names

dfs = [df, df1, df2]

sheets = ['df','df1','df2']   

 

# run function

dfs_tabs(dfs, sheets, 'filename.xlsx')

这是我如何将数据帧成功导出到 csv 到可以下载文件的位置的示例。希望我可以将下面的逻辑合并到上面的函数中，以便能够将文件下载到我的本地机器上。

#bounce to csv

outname = 'file.csv'

outdir = '/dbfs/FileStore/'

df.to_csv(outdir+outname, index=False, encoding="utf-8")

Answer 1

根据 https://docs.databricks.com/data/databricks-file-system.html#local-file-api-limitations - 我必须将整个本地地址附加到文件名变量 - 下面的完整工作脚本 - 感谢@jmcnamara

# function

def dfs_tabs(df_list, sheet_list, file_name):

    writer = pd.ExcelWriter(file_name,engine='xlsxwriter')  

    for dataframe, sheet in zip(df_list, sheet_list):

        dataframe.to_excel(writer, sheet_name=sheet, startrow=0 , startcol=0)  

    writer.save()

 

# list of dataframes and sheet names

dfs = [df, df1, df2]

sheets = ['df','df1','df2']   

 

# run function

dfs_tabs(dfs, sheets, '/local_disk0/tmp/filename.xlsx')

 

#copy file to dbfs

 

copyfile('/local_disk0/tmp/filename.xlsx', '/dbfs/FileStore/filename.xlsx')