Changeset 1698 in lmdz_wrf for trunk/tools/get_stations.py

Timestamp:

Dec 8, 2017, 1:51:42 PM (7 years ago)

Author:

lfita

Message:

Adding creation of 4D pressure variable

File:

• trunk/tools/get_stations.py (modified) (1 diff)

trunk/tools/get_stations.py

@@ -394 +394 @@
     dimvarvalues[dimvariables['lon']] = newXvarvals
     dimvarvalues[dimvariables['lat']] = newYvarvals
+    dx = newXvarvals.shape[1]
+    dy = newXvarvals.shape[0]
+
+# Looking for 4D 'plev' variable for calculation of diagnostics
+Zvarvals = dimvarvalues[dimvariables['plev']]
+Tvarvals = dimvarvalues[dimvariables['time']]
+if len(Zvarvals.shape) == 1:
+    dz = Zvarvals.shape[0]
+    dt = Tvarvals.shape[0]
+    newZvarvals = np.zeros((dt, dz, dy, dx), dtype=np.float)
+    for it in range(dt):
+        for iy in range(dy):
+            for ix in range(dx):
+                newZvarvals[it,:,iy,ix] = Zvarvals
+    dimvarvalues[dimvariables['plev']] = newZvarvals
 
 # Retrieving surface data