Changeset 688 in lmdz_wrf

Timestamp:

Jan 28, 2016, 2:47:49 PM (9 years ago)

Author:

lfita

Message:

Adding option for Ndimlon == 1 and dimx != dimy on `draw_points_lonlat'

File:

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

trunk/tools/drawing.py

@@ -2546 +2546 @@
     Ndimlon = len(olon.shape)
     if Ndimlon == 1:
-        lonvals = olon[:]
-        latvals = olat[:]
+        dx = olon.shape[0]
+        dy = olat.shape[0]
+        if dx == dy: 
+            lonvals = olon[:]
+            latvals = olat[:]
+        else: 
+            lonvals0 = np.zeros((dy,dx), dtype=np.float)
+            latvals0 = np.zeros((dy,dx), dtype=np.float)
+            for iL in range(dy):
+                lonvals0[iL,:] = olon[:]
+            for il in range(dx):
+                latvals0[:,il] = olat[:]
+            lonvals = lonvals0.flatten()
+            latvals = latvals0.flatten()
+
     elif Ndimlon == 2:
         lonvals = olon[:].flatten()