Changeset 512 for configs/testing/python/post_DCMIP31.py

Timestamp:

12/16/16 15:06:34 (8 years ago)

Author:

dubos

Message:

Testing : improved DCMIP31 params & plot

File:

• configs/testing/python/post_DCMIP31.py (modified) (3 diffs)

configs/testing/python/post_DCMIP31.py

@@ -13 +13 @@
     plt.savefig('T850.png')
 
-def plot_dT(nlon,nlat,llm, lon,T,Phi): # perturbation temp, final time
+def plot_dT(nlon,nlat,llm, lon,T,p,Phi): # perturbation temp, final time
     # vertical slice at final time
     print 'Reading data ...'
-    T, Phi =T[-1,:,nlat/2,:], Phi[-1,:,nlat/2,:]
+    T,p, Phi = T[-1,:,nlat/2,:], p[-1,:,nlat/2,:], Phi[-1,:,nlat/2,:]
     print '... done.'
     Cpd, kappa, g = 1004.5, 0.2857143, 9.80616
@@ -28 +28 @@
         lon2[lev,:] = lon[:]
 
+    theta = T*((peq/p)**kappa)
+    Thetab = Teq*np.exp(N2*z/g)
     Tb = G + (Teq-G)*np.exp(N2*z/g) # background temperature
     plt.figure(figsize=(12,6))
-    plt.contourf(lon2,z,T-Tb, levels=np.arange(-0.08,0.12,0.02) )
+    plt.contourf(lon2,z,theta-Thetab, levels=np.arange(-0.12,0.12,0.02) )
     plt.colorbar() 
-    plt.title('T-Tb')
+    plt.title('$\\Theta\'$')
     axis_longitude()
     plt.ylabel('z (m)')
@@ -41 +43 @@
 nc = cdf.Dataset(gridfile, "r")
 llm, nlon, nlat, ntime = getdims(nc, 'lev','lon','lat','time_counter')
-lon, lat, T850, T, Phi = getvars(nc, 'lon','lat','T850', 'T', 'PHI')
-plot_dT(nlon,nlat,llm, lon,T,Phi)
+lon, lat, T850, T, Phi, p = getvars(nc, 'lon','lat','T850', 'T', 'PHI','P')
+plot_dT(nlon,nlat,llm, lon,T,p,Phi)
 plot_T850(lon,lat,T850)