changed plot colors

acordonez · Dec 6, 2016 · 9c0b41a · 9c0b41a
1 parent 39a30f3
commit 9c0b41a
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 10 deletions.
diff --git a/make_seasonal_storm_composite_plots.py b/make_seasonal_storm_composite_plots.py
@@ -68,13 +68,16 @@
             ubox = ubox[k[0],:,:]
             vbox = vbox[k[0],:,:]
 
-            X,Y = np.meshgrid(range(0,50),range(0,50))
+            X,Y = np.meshgrid(range(0,101),range(0,101))
+
+            ubox = np.nanmean(ubox, axis = 0)
+            vbox = np.nanmean(vbox, axis = 0)
 
             f,axs = plt.subplots(1,1)
-            h = axs.pcolormesh(X,Y,np.nanmean(ibox,axis = 0),vmin = 0, vmax = 100)
-            axs.streamplot(X,Y,Umean,Vmean,linewidth = 1)
-            axs.contour(X,Y,np.nanmean(tbox,axis = 0),range(240,300,5))
-            axs.contour(X,Y,np.nanmean(box,axis = 0),range(96000,103000,100))
+            h = axs.pcolormesh(X,Y,np.nanmean(ibox,axis = 0),vmin = 0, vmax = 1)
+            axs.streamplot(X,Y,ubox,vbox,linewidth = 1)
+            axs.contour(X,Y,np.nanmean(tbox,axis = 0),range(240,300,5),colors = 'r')
+            axs.contour(X,Y,np.nanmean(box,axis = 0),range(96000,103000,100),colors = 'k')
             f.colorbar(h,ax = axs)
             f.savefig('test_' + str(n) + '_' + season + 'png')
 

diff --git a/make_storm_composite_plots.py b/make_storm_composite_plots.py
@@ -57,14 +57,15 @@
         ibox = ibox[k[0],:,:]
         ubox = ubox[k[0],:,:]
         vbox = vbox[k[0],:,:]
-
-        X,Y = np.meshgrid(range(0,50),range(0,50))
+        ubox = np.nanmean(ubox, axis = 0)
+        vbox = np.nanmean(vbox, axis = 0)
+        X,Y = np.meshgrid(range(0,101),range(0,101))
 
         f,axs = plt.subplots(1,1)
-        h = axs.pcolormesh(X,Y,np.nanmean(ibox,axis = 0),vmin = 0, vmax = 100)
+        h = axs.pcolormesh(X,Y,np.nanmean(ibox,axis = 0),vmin = 0, vmax = 1)
         axs.streamplot(X,Y,ubox,vbox,linewidth = 1)
-        axs.contour(X,Y,np.nanmean(tbox,axis = 0),range(240,300,5))
-        axs.contour(X,Y,np.nanmean(box,axis = 0),range(96000,103000,100))
+        axs.contour(X,Y,np.nanmean(tbox,axis = 0),range(240,300,5),colors = 'r')
+        axs.contour(X,Y,np.nanmean(box,axis = 0),range(96000,103000,100),colors = 'k')
         f.colorbar(h,ax = axs)
         f.savefig('test_' + str(n)+'.png')