diff --git a/src/cmsstyle/cmsstyle.py b/src/cmsstyle/cmsstyle.py index 7a23fc7..295f85f 100644 --- a/src/cmsstyle/cmsstyle.py +++ b/src/cmsstyle/cmsstyle.py @@ -2084,6 +2084,16 @@ def ylabel(self, label: str | None = None, labels: dict | None = None): elif labels is not None: for nframe in labels: self._frames[nframe].GetYaxis().SetTitle(labels[nframe]) + + def ylimits(self, limits: dict | None = None): + for nframe in limits: + self._frames[nframe].SetMinimum(limits[nframe][0]) + self._frames[nframe].SetMaximum(limits[nframe][1]) + #self._frames[nframe].GetYaxis().SetLimits(limits[nframe][0], limits[nframe][1]) + + def xlimits(self, limits: dict | None = None): + for nframe in limits: + self._frames[nframe].GetXaxis().SetLimits(limits[nframe][0], limits[nframe][1]) def save_figure(self, filename: str): self._canvas.SaveAs(filename)