From 11075f01cf629a6e14cac9637c87709f36af3ab6 Mon Sep 17 00:00:00 2001 From: Egor Tensin Date: Sat, 25 Jun 2016 05:52:06 +0300 Subject: scientific notation by default in plots --- algorithms/plotter.py | 11 +++++++++-- 1 file changed, 9 insertions(+), 2 deletions(-) (limited to 'algorithms/plotter.py') diff --git a/algorithms/plotter.py b/algorithms/plotter.py index 048894f..58ee941 100644 --- a/algorithms/plotter.py +++ b/algorithms/plotter.py @@ -13,6 +13,10 @@ class PlotBuilder: def set_ylabel(s): plt.ylabel(s) + @staticmethod + def set_yticklabels_scientific(): + plt.ticklabel_format(style='sci', axis='y', scilimits=(0, 0)) + @staticmethod def show_grid(): plt.grid() @@ -34,5 +38,8 @@ class PlotBuilder: plt.show() @staticmethod - def save(output_path): - plt.savefig(output_path)#, bbox_inches='tight') + def save(output_path, tight=False): + if tight: + plt.savefig(output_path, bbox_inches='tight') + else: + plt.savefig(output_path) -- cgit v1.2.3