Changeset 2026 in lmdz_wrf for trunk/tools/generic_tools.py

Timestamp:

Aug 6, 2018, 3:42:15 PM (7 years ago)

Author:

lfita

Message:

Adding:

• `stats_Slist': Function to provide extremes from a string as a list separated by a given character
• `inf_operSlist': Function to provide information from a string as a list separated by a given character followig a given operation and a set of values

File:

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

trunk/tools/generic_tools.py

@@ -97 +97 @@
 # getting_fixedline: Function to get the values from a line of text with fixed lenght of different values
 # grib_CFequiv: Function to provide the CF name of a GRIB variable code number
+# inf_operSlist: Function to provide information from a string as a list separated by 
+#   a given character followig a given operation and a set of values
 # ijlonlat: Function to provide the imin,jmin imax,jmax of a lon,lat box
 # incomming_flow: Function to determine if a fgrid-flow inflows to the central grid point
@@ -150 +152 @@
 # singleline_printing_class: Function to print all the values of a given class in a single line to be parseavel
 # stagger_unstagger: Function to de-stagger a variable
+# stats_Slist: Function to provide extremes from a string as a list separated by a given character
 # std_stats2Val: two variables standard Statistics class 
 # Str_Bool: Function to transform from a String value to a boolean one
@@ -13579 +13582 @@
 #create_LateX_figs('/home/lluis/estudios/FPS_ALPS/additional/IOP/analysis/figs', pltns, sts, 'png')
 
+def stats_Slist(listS, char=':'):
+    """ Function to provide extremes from a string as a list separated by a given 
+        character
+      listS= String as list
+      char= character to provide different values
+    >>> stats_Slist('1.2:3.45:-98.21:34321.1:100.1', ':')
+    stats_Slist min max mean standard_dev _______
+    stats_Slist_values: -98.21 34321.1 6865.528 13727.9292697
+    """
+    fname = 'stats_Slist'
+
+    haschar = listS.find(char)
+    if haschar == -1:
+        print errormsg
+        print '  ' + fname + ": provided String-list '" + + "' does not have " +     \
+          "character '" + char + "' !!"
+        quit(-1)
+
+    listv = str_list_k(listS, char, 'F')
+
+    nl = np.min(listv)
+    xl = np.max(listv)
+    ml = np.mean(listv)
+    sl = np.std(listv)
+
+    print fname + ' min max mean standard_dev _______'
+    print fname + '_values:', nl, xl, ml, sl
+
+    return
+
+#print stats_Slist('1.2:3.45:-98.21:34321.1:100.1', ':')
+
+def inf_operSlist(listS, oper, char=':', values=None):
+    """ Function to provide information from a string as a list separated by a given 
+        character followig a given operation and a set of values
+      listS= String as list
+      char= character to provide different values
+      oper= available operations 
+        'maxloc': index of the maximum value
+        'minloc': index of the minimum value
+        'sort': sort list of values (returned as a new String list)
+        'threshold': index of the values between which a given threshold is met (only 
+          useful for sort values)
+      values= char separated list of required values for the given operation (if 
+        applicable)
+    >>> inf_operSlist('1.2:3.45:-98.21:34321.1:100.1', 'sort', ':')
+    -98.21:1.2:3.45:100.1:34321.1
+    >>> inf_operSlist('1.2:3.45:-98.21:34321.1:100.1', 'minloc', ':')
+    2
+    >>> inf_operSlist('-98.21:1.2:3.45:100.1:34321.1', 'threshold', ':', 3.)
+    1
+    """
+    fname = 'inf_operSlist'
+    availoper = ['maxloc', 'minloc', 'sort', 'threshold']
+
+    haschar = listS.find(char)
+    if haschar == -1:
+        print errormsg
+        print '  ' + fname + ": provided String-list '" + + "' does not have " +     \
+          "character '" + char + "' !!"
+        quit(-1)
+
+    listv = str_list_k(listS, char, 'F')
+
+    if oper == 'maxloc':
+        xl = np.max(listv)
+        return index_vec(listv, xl)
+    elif oper == 'minloc':
+        nl = np.min(listv)
+        return index_vec(listv, nl)
+    elif oper == 'sort':
+        sortl = listv + []
+        sortl.sort()
+        sortlistS = ''
+        Nval = len(sortl)
+        sortlistS = ''
+        for iv in range(Nval-1):
+            sortlistS = sortlistS + str(sortl[iv]) + ':'
+        sortlistS = sortlistS + str(sortl[Nval-1])
+        if char != ':': return sortlistS.replace(':', char)
+        else: return sortlistS
+    elif oper == 'threshold':
+        if values is None:
+            print errormsg
+            print '  ' + fname + ": operation '" + oper + "' requires a value !!"
+            print "    None provided"
+            quit(-1)
+        val = np.float(values)
+        Nval = len(listv)
+        dlist = np.array(listv[1:Nval]) - np.array(listv[0:Nval-1])
+        if not np.all(dlist > 0.) or np.all(dlist < 0.): 
+            print errormsg
+            print '  ' + fname + ": operation '" + oper + "' requires sorted list !!"
+            print "    provided one '" + listS + "' is not!!"
+            quit(-1)
+        sign = listv[1] - listv[0]
+        if sign > 0.:
+            for iloc in range(Nval-1):
+                if listv[iloc+1] >= val and listv[iloc] < val: return iloc
+            return Nval
+        elif sign < 0.:
+            for iloc in range(Nval-1):
+                if listv[iloc] >= val and listv[iloc-1] < val: return iloc
+            return Nval
+
+    else:
+        print errormsg
+        print '  ' + fname + ": operation '" + oper + "' nor ready !!"
+        print "    available ones:", availoper
+        quit(-1)
+
+    return       
+
 #quit()