same axis update

plotter/Plotter.py

@@ -326,8 +326,6 @@ class Plotter:
             else:
                 min_val = {var:self.min_y[i] for var, i in zip(self.var, list(range(len(self.var))))}
 
-            # print(max_val)
-
             duration = self.ani_period * len(self.filename)
 
             i = 0
@@ -341,13 +339,15 @@ class Plotter:
                 png_names = []
 
                 for data in self.file_data:
+                    levels = np.linspace(min_val[var], max_val[var], 30)
+
                     fig,ax=plt.subplots(1,1)
                     ax.set_title(title)
                     ax.set_xlabel(x_name)
                     ax.set_ylabel(y_name)
 
                     Z = data.data[var]
-                    cp = ax.contourf(X, Y, Z, vmin=min_val[var], vmax=max_val[var])
+                    cp = ax.contourf(X, Y, Z, vmin=min_val[var], vmax=max_val[var], levels=levels)
                     fig.colorbar(cp) # Add a colorbar to a plot
                     plt.close(fig)
                     figname = var + str(counter)