Add option to remove axes from the figures

orbingol · Feb 16, 2018 · b6f28a4 · b6f28a4
1 parent 6b1301f
commit b6f28a4
Showing 1 changed file with 24 additions and 0 deletions.
diff --git a/geomdl/visualization/VisMPL.py b/geomdl/visualization/VisMPL.py
@@ -74,6 +74,10 @@ def render(self):
         if self._config.display_legend:
             plt.legend(legend_proxy, legend_names)
 
+        # Remove axes
+        if not self._config.display_axes:
+            plt.axis('off')
+
         # Display 2D plot
         plt.show()
 
@@ -120,6 +124,10 @@ def render(self):
         if self._config.display_legend:
             ax.legend(legend_proxy, legend_names, numpoints=1)
 
+        # Remove axes
+        if not self._config.display_axes:
+            plt.axis('off')
+
         # Display the 3D plot
         plt.show()
 
@@ -165,6 +173,10 @@ def render(self):
         if self._config.display_legend:
             ax.legend(legend_proxy, legend_names, numpoints=1)
 
+        # Remove axes
+        if not self._config.display_axes:
+            plt.axis('off')
+
         # Display the 3D plot
         plt.show()
 
@@ -210,6 +222,10 @@ def render(self):
         if self._config.display_legend:
             ax.legend(legend_proxy, legend_names, numpoints=1)
 
+        # Remove axes
+        if not self._config.display_axes:
+            plt.axis('off')
+
         # Display the 3D plot
         plt.show()
 
@@ -255,6 +271,10 @@ def render(self):
         if self._config.display_legend:
             ax.legend(legend_proxy, legend_names, numpoints=1)
 
+        # Remove axes
+        if not self._config.display_axes:
+            plt.axis('off')
+
         # Display the 3D plot
         plt.show()
 
@@ -300,5 +320,9 @@ def render(self):
         if self._config.display_legend:
             ax.legend(legend_proxy, legend_names, numpoints=1)
 
+        # Remove axes
+        if not self._config.display_axes:
+            plt.axis('off')
+
         # Display the 3D plot
         plt.show()