Py plot: correct vertical extension of color legend

Too high in rectangular images, should align with vertical extension of main frame.

Edited by Joachim Wuttke