Changeset 395 for trunk/UTIL/PYTHON/myplot.py

Timestamp:

Nov 17, 2011, 1:08:01 PM (13 years ago)

Author:

acolaitis

Message:

PYTHON

M 393 mymath.py
^{----------------- Cosmetic change}

M 393 make_netcdf.py
^{----------------- Cosmetic change}

M 393 planetoplot.py
^{----------------- corrected bug with varname and --tsat}

M 393 myplot.py
^{----------------- added possibility for the script to read ncdf files with NaN value and treat them correctly without messing with the mathematical operations.}

File:

• trunk/UTIL/PYTHON/myplot.py (modified) (3 diffs)

trunk/UTIL/PYTHON/myplot.py

@@ -49 +49 @@
 def getfield (nc,var):
     ## this allows to get much faster (than simply referring to nc.variables[var])
+    import numpy as np
     dimension = len(nc.variables[var].dimensions)
     #print "   Opening variable with", dimension, "dimensions ..."
@@ -54 +55 @@
     elif dimension == 3:  field = nc.variables[var][:,:,:]
     elif dimension == 4:  field = nc.variables[var][:,:,:,:]
-    return field
+    # if there are NaNs in the ncdf, they should be loaded as a masked array which will be
+    # recasted as a regular array later in reducefield
+    if (np.isnan(np.sum(field)) and (type(field).__name__ not in 'MaskedArray')):
+       print "Warning: netcdf as nan values but is not loaded as a Masked Array."
+       print "recasting array type"
+       out=np.ma.masked_invalid(field)
+       out.set_fill_value([np.NaN])
+    else:
+       out=field
+    return out
 
 ## Author: AS + TN + AC
@@ -492 +502 @@
                               llcrnrlat=wlat[0],urcrnrlat=wlat[1],llcrnrlon=wlon[0],urcrnrlon=wlon[1])
     elif char == "npstere": m = Basemap(rsphere=radius,projection='npstere', boundinglat=blat, lon_0=0.)
-    elif char == "spstere": m = Basemap(rsphere=radius,projection='spstere', boundinglat=blat, lon_0=0.)
+    elif char == "spstere": m = Basemap(rsphere=radius,projection='spstere', boundinglat=blat, lon_0=180.)
     elif char == "nplaea":  m = Basemap(rsphere=radius,projection='nplaea', boundinglat=wlat[0], lon_0=meanlon)
     elif char == "laea":    m = Basemap(rsphere=radius,projection='laea',lon_0=meanlon,lat_0=meanlat,lat_ts=meanlat,\