Preserve row selections when sorting `st.dataframe` on the frontend

[sfc-gh-dmatthews]

Checklist

• [X] I have searched the existing issues for similar feature requests.
• [X] I added a descriptive title and summary to this issue.

Summary

When using st.dataframe with row selections (and not column selections), sorting the dataframe will clear away the row selections. Users can (partially) get around the issue using the search function in the dataframe toolbar which preserves row selections to make working with larger dataframes feasible.

Why?

Although the search bar provides a workaround when handling large datasets, it's a very natural thing for a user to want to sort a dataframe to get to rows they want to select. Currently, they need to be told (or discover frustratingly) that they can't sort in the middle of making row selections.

How?

Preserve the selection during frontend manipulation.

Additional Context

No response

[github-actions[bot]]

To help Streamlit prioritize this feature, react with a 👍 (thumbs up emoji) to the initial post.

Your vote helps us identify which enhancements matter most to our users.

[nimeshrevo]

Also, is there a way to preserve the selection state, when the table is being updated. For example: I have a st.dataframe with selection_mode = "multi-row" inside a function with experimental_fragment decorator, as I need to update the table on certain interval. As, a user if I select row(s), they get deselected while the table gets updated. Is there any work around for this? Any help is appreciated, Thank you!