I'm not sure how to see only individual staffs on the keyboard during playback.
The next "problem" is that I would also like to see the notes I'm actually inputting or currently clicking. I'm having trouble with certain cleffs :) I've also had a problem I'm presenting you in the example. So, I'm clicking on the note, hearing something weird, and looking for MuseScore keyboard to help me out, and visualise what it is trying to do. I'm not a great musician, and you can imagine my confusion doing this with lots of accidentals in Alto cleff.