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

Timestamp:

May 24, 2016, 11:05:49 AM (9 years ago)

Author:

lfita

Message:

Adding draw_vertical_levels' and plot_vertical_lev', plotting vertical levels distribution

File:

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

trunk/tools/drawing_tools.py

@@ -6561 +6561 @@
     return
 
+
+def plot_vertical_lev(vertz, zlog, dzlog, gtit, kfig, lloc):
+    """ plotting vertical levels distribution
+    plot_vertical_lev(vertz, gtit, kfig, lloc)
+      vertz= distribution of vertical heights [z]
+      zlog: to use logarithmic scale on the height axis ('true/false')
+      dzlog: to use logarithmic scale on the difference of height between levels axis ('true/false')
+      gtit= title of the graph ('!' for spaces)
+      kfig= kind of figure (jpg, pdf, png)
+      lloc= location of the legend (0, automatic)
+        1: 'upper right', 2: 'upper left', 3: 'lower left', 4: 'lower right',
+        5: 'right', 6: 'center left', 7: 'center right', 8: 'lower center',
+        9: 'upper center', 10: 'center'      kfig= kind of figure
+    """
+    fname = 'plot_vertical_lev'
+
+    figname = 'vertical_lev'
+
+    if vertz == 'h':
+        print fname + '_____________________________________________________________'
+        print plot_vertical_lev.__doc__
+        quit()
+
+    Nlev = len(vertz)
+    dvertz = vertz[1:Nlev] - vertz[0:Nlev-1]
+
+# From: http://stackoverflow.com/questions/14762181/adding-a-y-axis-label-to-secondary-y-axis-in-matplotlib
+
+    plt.rc('text', usetex=True)
+
+    fig, ax1 = plt.subplots()
+    ax2 = ax1.twinx()
+
+    plt.xlim(0,Nlev)
+
+    l1 = ax1.plot(range(Nlev), vertz, 'r-x', label='height')
+    l2 = ax2.plot(range(1,Nlev), dvertz, 'b-x', label='dheight')
+
+    ax1.set_xlabel('level (\#)')
+    ax1.set_ylabel('height (m)', color='r')
+    ax1.set_ylim(1,np.max(vertz))
+    ax1.grid()
+    ax2.set_ylabel('difference between levels (m)', color='b')
+
+    plt.title(gtit)
+    if zlog: ax1.set_yscale('log')
+    if dzlog: ax2.set_yscale('log')
+
+    output_kind(kfig, figname, True)
+
+    return
+