How to save DataFrame as csv in streamlit?

Question

Hi I try to save my DataFrame as csv file in streamlit. My program return two Dataframes to streamlit as data. I want to make button that will allow to save this dataframes. I try this simple code:

if st.button("Pobierz plik"):
    data[0].to_csv('file.csv',index=False)

But nothing hapend when i clicked on the button. Any one have some idea?

Do you want to save the file locally or do you want it to be downloaded from Streamlit? — Ailurophile
– Ailurophile, Commented Mar 29, 2022 at 10:36

Ailurophile · Accepted Answer · 2022-03-29 10:58:36Z

2

You need to use st.download_button

For streamlit > 1.0.0

Display a download button widget.

This is useful when you would like to provide a way for your users to download a file directly from your app.

Note that the data to be downloaded is stored in-memory while the user is connected, so it's a good idea to keep file sizes under a couple hundred megabytes to conserve memory.

A working example can be found here

In your case

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

csv = convert_df(data[0])

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

answered Mar 29, 2022 at 10:58

Ailurophile

3,02510 gold badges33 silver badges65 bronze badges

Sign up to request clarification or add additional context in comments.

1 Comment

Agustin Martin Over a year ago

Working wonders, added some args to the df.to_csv command to personalize it even more. Thank you very much !

Collectives™ on Stack Overflow

How to save DataFrame as csv in streamlit?

1 Answer 1

1 Comment

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

1 Comment

Your Answer

Sign up or log in

Post as a guest

Related