Changeset 1573 in lmdz_wrf for trunk/tools/drawing_tools.py

Timestamp:

May 12, 2017, 9:25:55 PM (8 years ago)

Author:

lfita

Message:

Adding use of 'auto'
Changing example

File:

• trunk/tools/drawing_tools.py (modified) (3 diffs)

trunk/tools/drawing_tools.py

@@ -9897 +9897 @@
       ifclose= whether figure should be closed
     """
+    from scipy import stats as sts
     fname = 'plot_time_lag'
 
     Nvalues = len(prevals)
 
-    dxn = gen.fillValue
-    dxx = -gen.fillValue
-    dyn = gen.fillValue
-    dyx = -gen.fillValue
-
-    for iv in range(Nvalues):
-        minv = np.min(prevals[iv])
-        maxv = np.max(prevals[iv])
-        if minv < dxn: dxn = minv
-        if maxv > dxx: dxx = maxv
-        minv = np.min(futvals[iv])
-        maxv = np.max(futvals[iv])
-        if minv < dyn: dyn = minv
-        if maxv > dyx: dyx = maxv
+    dxn = prerange[0]
+    dxx = prerange[1]
+    dyn = futrange[0]
+    dyx = futrange[1]
 
     axstyles = ['pretty', 'Nfix', 'Stringlist', 'Vfix']
@@ -10073 +10064 @@
 
             Nlag = int(figv[1])
-            col = gen.str_list_rep(figv[2],'@',Nvalues)
+            if figv[2] == 'auto':
+                col = colorsauto
+            else:
+                col = gen.str_list_rep(figv[2],'@',Nvalues)
             mark = gen.str_list_rep(figv[3],'@',Nvalues)
             size = gen.str_list_rep(figv[4],'@',Nvalues)
@@ -10083 +10077 @@
                 plt.plot(xvs[0:dimt-Nlag], yvs[Nlag:dimt], ',', color=col[iv],       \
                   marker=mark[iv], markersize=size[iv])
+            # Computing correlation
+            corr, p_value = sts.pearsonr(xvs[0:dimt-Nlag], yvs[Nlag:dimt])
+            txt='corr= ' + '{0:.3g}'.format(corr) + ' p: ' + '{0:.2g}'.format(p_value)
+            plt.annotate(txt,xy=(0.95,0.05*(Nvalues-iv)), xycoords='figure fraction',\
+              color=col[iv], fontsize=lsiz, horizontalalignment='right')
         else:
             print errormsg