【Streamlit学习笔记】实现包含多个sheet的excel文件下载

1、什么是Streamlit

Streamlit是一个免费的开源框架，用于快速构建和共享漂亮的机器学习和数据科学Web应用程序，官网链接 Streamlit
Streamlit API链接 API reference
实际项目中遇到的问题：包含多个sheet的excel文件下载，下面将给出实现程序

2、st.download_button下载excel文件

官网给出的参考程序，下载csv文件例程如下：

import streamlit as st

@st.cache_data
def convert_df(df):
    # IMPORTANT: Cache the conversion to prevent computation on every rerun
    return df.to_csv().encode('utf-8')

csv = convert_df(my_large_df)

st.download_button(
    label="Download data as CSV",
    data=csv,
    file_name='large_df.csv',
    mime='text/csv',
)

如上所述程序，测试发现无法下载包含多个sheet的excel文件

3、st.download_button下载包含多个sheet的excel文件

废话不多说，直接给出程序：

from io import BytesIO
import streamlit as st
import pandas as pd

xlsx_files_path = 'excel文件路径'
df = pd.read_excel(xlsx_files_path,sheet_name=None,header=0,index_col=0)
excel_keys = list(df.keys())
output = BytesIO()
writer = pd.ExcelWriter(output, engine='xlsxwriter')
for k in range(len(excel_keys)):
	df = pd.read_excel(xlsx_files_path,sheet_name=excel_keys[k],header=0,index_col=0)
	df.to_excel(writer, sheet_name=excel_keys[k])
writer.close()
st.download_button('📥下载文件至本地', data = output.getvalue(), file_name = 'excel文件名', mime="application/vnd.ms-excel")

亲测有效，下载成功！！！效果如图所示：

在这里插入图片描述

4、多个sheet的excel文件非常大的情况

如果需要下载的excel文件非常大，上述程序每次加载会非常慢，现对其进行优化

首先使用st.cache_data的方式，但是其只能解决首次加载后，可快速加载，文件的首次加载仍然耗时严重

from io import BytesIO
import streamlit as st
import pandas as pd

@st.cache_data
def output_xlsx(xlsx_files_path):
    df = pd.read_excel(xlsx_files_path,sheet_name=None,header=0,index_col=0)
    excel_keys = list(df.keys())
    output = BytesIO()
    writer = pd.ExcelWriter(output, engine='xlsxwriter')
    for k in range(len(excel_keys)):
        df = pd.read_excel(xlsx_files_path,sheet_name=excel_keys[k],header=0,index_col=0)
        df.to_excel(writer, sheet_name=excel_keys[k])
    writer.close()
    return output
xlsx_files_path = 'excel文件路径'
output = output_xlsx(xlsx_files_path)
st.download_button('📥下载文件至本地', data = output.getvalue(), file_name = 'excel文件名', mime="application/vnd.ms-excel")

如果想excel文件首次即快速下载至本地，可换种思路，通过zip文件下载

import streamlit as st
import zipfile

xlsx_files_path = 'excel文件路径'
zip_file = zipfile.ZipFile(xlsx_files_path.split('.xlsx')[0]+'.zip','w')
zip_file.write(xlsx_files_path,'excel文件名')
zip_file.close()
zip_file_data = open(xlsx_files_path.split('.xlsx')[0]+'.zip', "rb")
with zip_file_data as fp:
	download_flag = st.download_button('📥下载文件至本地', data = fp, file_name = 'excel文件名'+'.zip', mime="application/zip")
zip_file_data.close()