如何在 Streamlit 中将 Pandas DataFrame 下载到 CSV 文件

Question

我有一个大型 DataFrame，我想在 Excel 中探索并转移到其他应用程序。 如何在我的 Streamlit 应用程序中将其下载为 CSV 文件？

"

Answer 1

从 docs ，假设您有一个 DataFrame my_large_df ，您可以编写一个函数：

import streamlit as st

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


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

Answer 2

import streamlit as st

my_large_df = pd.read_csv('file.csv') #This is your 'my_large_df'

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

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