plot_utils: Make post_plot_setup() set the xlim
diff --git a/cr2/plot_utils.py b/cr2/plot_utils.py
index e361fa6..45a9606 100644
--- a/cr2/plot_utils.py
+++ b/cr2/plot_utils.py
@@ -47,8 +47,8 @@
return ax
-def post_plot_setup(ax, title="", xlabel=None, ylim=None):
- """Set xlabel, title and ylim of the plot
+def post_plot_setup(ax, title="", xlabel=None, xlim=None, ylim=None):
+ """Set xlabel, title, xlim adn ylim of the plot
This has to be called after calls to .plot()
"""
@@ -67,6 +67,9 @@
ax.set_ylim(ylim[0], ylim[1])
+ if xlim:
+ ax.set_xlim(xlim[0], xlim[1])
+
def plot_allfreqs(in_power, out_power, map_label, title="", width=None, height=None):
"""Do allfreqs plots similar to those of CompareRuns"""
import power