BUG: when a table is sorted using "sort ascending/descending", the row selections do not move to the new row positions

## Summary

When a table is sorted with the "sort ascending/descending" items in the context menu of a column, the row selections should update in a consistent manner.

However, after sorting the row selections stay in their old order, so the wrong set of rows are then selected.

## Example

Using the example on , suppose the first row is selected, in this case Gabriela:


If one clicks "Sort ascending" on a column, Gabriela should should remain selected, but now a different person is selected:


Related issue in streamlit: https://github.com/streamlit/streamlit/issues/11345