Changeset 398 for trunk/UTIL/PYTHON/mymath.py

Timestamp:

Nov 18, 2011, 1:51:41 PM (13 years ago)

Author:

acolaitis

Message:

PYTHON:

• Improved handling of missing values NaN by min() and max() functions
• Added the possibility to change the viewing angle of "ortho" projections by using --blat. Default is meanlat. (remember than in spstere and npstere projections, blat already controls the boundling latitude)
• Changed colorbar of "operation" plots to RdBu_r, whatever the colorbar of other plots are. The thinking behind this is that this is a symmetrical colorbar, which is very usefull for difference plots. (could be changed for other operations.)

File:

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

trunk/UTIL/PYTHON/mymath.py

@@ -2 +2 @@
         import numpy as np
         if field is None: return None
+        if type(field).__name__=='MaskedArray':
+              field.set_fill_value(np.NaN)
+              return np.ma.array(field).min(axis=axis)
+        elif (np.isnan(np.sum(field)) and (type(field).__name__ not in 'MaskedArray')):
+              return np.ma.masked_invalid(field).min(axis=axis)
         else: return np.array(field).min(axis=axis)
 
@@ -7 +12 @@
         import numpy as np
         if field is None: return None
+        if type(field).__name__=='MaskedArray':
+              field.set_fill_value(np.NaN)
+              return np.ma.array(field).max(axis=axis)
+        elif (np.isnan(np.sum(field)) and (type(field).__name__ not in 'MaskedArray')):
+              return np.ma.masked_invalid(field).max(axis=axis) 
         else: return np.array(field).max(axis=axis)
 
@@ -16 +26 @@
               field.set_fill_value(np.NaN)
               zout=np.ma.array(field).mean(axis=axis)
-#              np.ma.masked_invalid(zout)
               zout.set_fill_value(np.NaN)
+              return zout.filled()
+           elif (np.isnan(np.sum(field)) and (type(field).__name__ not in 'MaskedArray')):
+              zout=np.ma.masked_invalid(field).mean(axis=axis)
+              zout.set_fill_value([np.NaN])
               return zout.filled()
            else: