This would not be too hard with the Dash data table. Specifically, see this section of the docs about the active_cell property. You could use that as a State value for your callback, and the button as the Input. It should work just like you’re describing.
CLICK HERE to find out more related problems solutions.