diff --git a/electrode_check.py b/electrode_check.py index 46e05f0..4ed71ed 100644 --- a/electrode_check.py +++ b/electrode_check.py @@ -50,6 +50,13 @@ class plot(): def keypress(self, event): if event.key == 'v': + yrange = np.abs(np.diff(self.axs[0].get_ylim())) + for ax in self.axs: + ax.set_ylim(-(yrange / 2) * (2/3), (yrange / 2) * (2/3) ) + if event.key == 'V': + yrange = np.abs(np.diff(self.axs[0].get_ylim())) + for ax in self.axs: + ax.set_ylim(-(yrange / 2) * (3/2), (yrange / 2) * (3/2) ) print('v pressed')