Improve examples: delete pdfs, tweak example.py

Delete PDF output to have a cleaner figures folder

examples/plot-examples.py

@@ -4,6 +4,16 @@
 import matplotlib.pyplot as plt
 import scienceplots
 
+import os
+
+# Check we are in examples dir
+current_dir = os.getcwd().lower()
+if (current_dir.endswith('scienceplots')):
+    os.chdir('./examples')
+# Create 'figures' folder if it does not exist
+if (not os.path.exists('./figures')):
+    os.makedirs('figures')
+
 def model(x, p):
     return x ** (2 * p + 1) / (1 + x ** (2 * p))
 
@@ -19,8 +29,8 @@ def model(x, p):
     ax.legend(title='Order')
     ax.autoscale(tight=True)
     ax.set(**pparam)
-    fig.savefig('figures/fig1.pdf')
-    fig.savefig('figures/fig1.jpg', dpi=300)
+    fig.savefig('figures/fig01.jpg', dpi=300)
+    plt.close()
 
 with plt.style.context(['science', 'ieee']):
     fig, ax = plt.subplots()
@@ -29,10 +39,8 @@ def model(x, p):
     ax.legend(title='Order')
     ax.autoscale(tight=True)
     ax.set(**pparam)
-    # Note: $\mu$ doesn't work with Times font (used by ieee style)
-    ax.set_ylabel(r'Current (\textmu A)')  
-    fig.savefig('figures/fig2a.pdf')
-    fig.savefig('figures/fig2a.jpg', dpi=300)
+    fig.savefig('figures/fig02a.jpg', dpi=300)
+    plt.close()
 
 with plt.style.context(['science', 'ieee', 'std-colors']):
     fig, ax = plt.subplots()
@@ -41,10 +49,8 @@ def model(x, p):
     ax.legend(title='Order')
     ax.autoscale(tight=True)
     ax.set(**pparam)
-    # Note: $\mu$ doesn't work with Times font (used by ieee style)
-    ax.set_ylabel(r'Current (\textmu A)')  
-    fig.savefig('figures/fig2b.pdf')
-    fig.savefig('figures/fig2b.jpg', dpi=300)
+    fig.savefig('figures/fig02b.jpg', dpi=300)
+    plt.close()
 
 with plt.style.context(['science', 'nature']):
     fig, ax = plt.subplots()
@@ -53,8 +59,8 @@ def model(x, p):
     ax.legend(title='Order')
     ax.autoscale(tight=True)
     ax.set(**pparam)
-    fig.savefig('figures/fig2c.pdf')
-    fig.savefig('figures/fig2c.jpg', dpi=300)
+    fig.savefig('figures/fig02c.jpg', dpi=300)
+    plt.close()
 
 with plt.style.context(['science', 'scatter']):
     fig, ax = plt.subplots(figsize=(4, 4))
@@ -73,8 +79,8 @@ def model(x, p):
     ax.set_ylabel(ylbl)
     ax.set_xlim([-2, 2])
     ax.set_ylim([-2, 2])
-    fig.savefig('figures/fig3.pdf')
-    fig.savefig('figures/fig3.jpg', dpi=300)
+    fig.savefig('figures/fig03.jpg', dpi=300)
+    plt.close()
 
 with plt.style.context(['science', 'high-vis']):
     fig, ax = plt.subplots()
@@ -83,8 +89,8 @@ def model(x, p):
     ax.legend(title='Order')
     ax.autoscale(tight=True)
     ax.set(**pparam)
-    fig.savefig('figures/fig4.pdf')
-    fig.savefig('figures/fig4.jpg', dpi=300)
+    fig.savefig('figures/fig04.jpg', dpi=300)
+    plt.close()
 
 with plt.style.context(['dark_background', 'science', 'high-vis']):
     fig, ax = plt.subplots()
@@ -93,8 +99,8 @@ def model(x, p):
     ax.legend(title='Order')
     ax.autoscale(tight=True)
     ax.set(**pparam)
-    fig.savefig('figures/fig5.pdf')
-    fig.savefig('figures/fig5.jpg', dpi=300)
+    fig.savefig('figures/fig05.jpg', dpi=300)
+    plt.close()
 
 with plt.style.context(['science', 'notebook']):
     fig, ax = plt.subplots()
@@ -103,8 +109,8 @@ def model(x, p):
     ax.legend(title='Order')
     ax.autoscale(tight=True)
     ax.set(**pparam)
-    fig.savefig('figures/fig10.pdf')
     fig.savefig('figures/fig10.jpg', dpi=300)
+    plt.close()
 
 # Plot different color cycles
 
@@ -115,8 +121,8 @@ def model(x, p):
     ax.legend(title='Order')
     ax.autoscale(tight=True)
     ax.set(**pparam)
-    fig.savefig('figures/fig6.pdf')
-    fig.savefig('figures/fig6.jpg', dpi=300)
+    fig.savefig('figures/fig06.jpg', dpi=300)
+    plt.close()
 
 with plt.style.context(['science', 'vibrant']):
     fig, ax = plt.subplots()
@@ -125,8 +131,8 @@ def model(x, p):
     ax.legend(title='Order')
     ax.autoscale(tight=True)
     ax.set(**pparam)
-    fig.savefig('figures/fig7.pdf')
-    fig.savefig('figures/fig7.jpg', dpi=300)
+    fig.savefig('figures/fig07.jpg', dpi=300)
+    plt.close()
 
 with plt.style.context(['science', 'muted']):
     fig, ax = plt.subplots()
@@ -135,8 +141,8 @@ def model(x, p):
     ax.legend(title='Order', fontsize=7)
     ax.autoscale(tight=True)
     ax.set(**pparam)
-    fig.savefig('figures/fig8.pdf')
-    fig.savefig('figures/fig8.jpg', dpi=300)
+    fig.savefig('figures/fig08.jpg', dpi=300)
+    plt.close()
 
 with plt.style.context(['science', 'retro']):
     fig, ax = plt.subplots()
@@ -145,8 +151,8 @@ def model(x, p):
     ax.legend(title='Order')
     ax.autoscale(tight=True)
     ax.set(**pparam)
-    fig.savefig('figures/fig9.pdf')
-    fig.savefig('figures/fig9.jpg', dpi=300)
+    fig.savefig('figures/fig09.jpg', dpi=300)
+    plt.close()
 
 with plt.style.context(['science', 'grid']):
     fig, ax = plt.subplots()
@@ -155,8 +161,8 @@ def model(x, p):
     ax.legend(title='Order')
     ax.autoscale(tight=True)
     ax.set(**pparam)
-    fig.savefig('figures/fig11.pdf')
     fig.savefig('figures/fig11.jpg', dpi=300)
+    plt.close()
 
 with plt.style.context(['science', 'high-contrast']):
     fig, ax = plt.subplots()
@@ -165,8 +171,8 @@ def model(x, p):
     ax.legend(title='Order')
     ax.autoscale(tight=True)
     ax.set(**pparam)
-    fig.savefig('figures/fig12.pdf')
     fig.savefig('figures/fig12.jpg', dpi=300)
+    plt.close()
 
 with plt.style.context(['science', 'light']):
     fig, ax = plt.subplots()
@@ -175,8 +181,8 @@ def model(x, p):
     ax.legend(title='Order', fontsize=7)
     ax.autoscale(tight=True)
     ax.set(**pparam)
-    fig.savefig('figures/fig13.pdf')
     fig.savefig('figures/fig13.jpg', dpi=300)
+    plt.close()
 
 # Note: You need to install the Noto Serif CJK Fonts before running 
 # examples 14 and 15. See FAQ in README.
@@ -190,6 +196,7 @@ def model(x, p):
     ax.set(ylabel=r'電流 ($\mu$A)')
     ax.autoscale(tight=True)
     fig.savefig('figures/fig14a.jpg', dpi=300)
+    plt.close()
 
 with plt.style.context(['science', 'no-latex', 'cjk-sc-font']):
     fig, ax = plt.subplots()
@@ -200,6 +207,7 @@ def model(x, p):
     ax.set(ylabel=r'电流 ($\mu$A)')
     ax.autoscale(tight=True)
     fig.savefig('figures/fig14b.jpg', dpi=300)
+    plt.close()
 
 with plt.style.context(['science', 'no-latex', 'cjk-jp-font']):
     fig, ax = plt.subplots()
@@ -210,6 +218,7 @@ def model(x, p):
     ax.set(ylabel=r'電気 ($\mu$A)')
     ax.autoscale(tight=True)
     fig.savefig('figures/fig14c.jpg', dpi=300)
+    plt.close()
 
 with plt.style.context(['science', 'no-latex', 'cjk-kr-font']):
     fig, ax = plt.subplots()
@@ -220,6 +229,7 @@ def model(x, p):
     ax.set(ylabel=r'전류 ($\mu$A)')
     ax.autoscale(tight=True)
     fig.savefig('figures/fig14d.jpg', dpi=300)
+    plt.close()
 
 # import matplotlib
 # matplotlib.use('pgf')  # stwich backend to pgf
@@ -243,6 +253,7 @@ def model(x, p):
 #     ax.set(ylabel=r'電流 ($\mu$A)')
 #     ax.autoscale(tight=True)
 #     fig.savefig('figures/fig15.pdf', backend='pgf')
+#     plt.close()
 
 with plt.style.context(['science', 'russian-font']):
     fig, ax = plt.subplots()
@@ -253,6 +264,7 @@ def model(x, p):
     ax.set(ylabel=r'Сила тока ($\mu$A)')
     ax.autoscale(tight=True)
     fig.savefig('figures/fig16.jpg', dpi=300)
+    plt.close()
 
 with plt.style.context(['science', 'turkish-font']):
     fig, ax = plt.subplots()
@@ -263,3 +275,4 @@ def model(x, p):
     ax.set(ylabel=r'Mevcut Güç/Akım ($\mu$A)')
     ax.autoscale(tight=True)
     fig.savefig('figures/fig17.jpg', dpi=300)
+    plt.close()