如何将熊猫DataFrame下载到Streamlit中的CSV文件

pip install pandas streamlit

import pandas as pd
import streamlit as st

# 假设已经有一个名为df的熊猫DataFrame
df = pd.DataFrame(...)

# 创建下载链接
def create_download_link(df, filename):
    csv = df.to_csv(index=False)
    b64 = base64.b64encode(csv.encode()).decode()  # 将CSV文件编码为base64字符串
    href = f'<a href="data:file/csv;base64,{b64}" download="{filename}.csv">点击此处下载CSV文件</a>'
    return href

# 在Streamlit应用程序中显示下载链接
st.markdown(create_download_link(df, 'dataframe'), unsafe_allow_html=True)

import base64