python - 如何在python和streamlit中下载excel文件？

Question

我有一个 Python 脚本，它可以读取文件并使用 Python 和 streamlit 将其转换为数据帧。然后我想创建一个函数，允许用户将此数据帧下载为扩展名为 .xls 的 Excel 文件。 .
所以我尝试读取数据框并使用这两个函数将其转换为 Excel 文件:

pd.ExcelWriter
df.to_excel

但是当我尝试使用链接下载文件时，文件不会下载并显示此错误:

Failed-Network error

代码:

import pandas as pd 
import streamlit as st

writer = pd.ExcelWriter('update2.xlsx')
df.to_excel(writer, index = False, header=True,encoding='utf-8')
with open(writer,'rb') as f:
    b64 = base64.b64encode(f.read())
    href = f'<a href="data:file/xls;base64,{b64}" download="new_file.{extension}">Download {extension}</a>'

st.write(href, unsafe_allow_html=True)

Answer 1

使用 streamlit 最新版本(1.0.0 以上):
利用

st.download_button

显示下载按钮小部件。
当您希望为用户提供一种直接从您的应用下载文件的方式时，这很有用。
请注意，要下载的数据在用户连接时存储在内存中，因此最好将文件大小保持在几百兆字节以下以节省内存。
这是来自 discussion 的示例代码，这对下载excel文件很有帮助...

import pandas as pd
from io import BytesIO
from pyxlsb import open_workbook as open_xlsb
import streamlit as st

def to_excel(df):
    output = BytesIO()
    writer = pd.ExcelWriter(output, engine='xlsxwriter')
    df.to_excel(writer, index=False, sheet_name='Sheet1')
    workbook = writer.book
    worksheet = writer.sheets['Sheet1']
    format1 = workbook.add_format({'num_format': '0.00'}) 
    worksheet.set_column('A:A', None, format1)  
    writer.save()
    processed_data = output.getvalue()
    return processed_data
df_xlsx = to_excel(df)
st.download_button(label='📥 Download Current Result',
                                data=df_xlsx ,
                                file_name= 'df_test.xlsx')