scientific notation by default in plots

1 files changed, 9 insertions, 2 deletions

diff --git a/algorithms/plotter.py b/algorithms/plotter.py
index 048894f..58ee941 100644
--- a/algorithms/plotter.py
+++ b/algorithms/plotter.py
@@ -14,6 +14,10 @@ class PlotBuilder:
         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)