Ejemplo n.º 1
0
def zr_density_pdf(ax,Z,R,title=None,grid=(80,60),minct=0.01,
                   xlabFontSize=16,xpad=7,ylabFontSize=16,ypad=7,
                   xlims=(-3,3),ylims=(-20.,60.)):
    """Create a density scatterplot using the hex bin method (color marker based on density)
 INPUT::
  ax              = Axis to plot against (produced externally)
  Z               = Reflectivity factor [mm^6/m^3]
  R               = Rainfall rate [mm/h]
     OPTIONAL
  title           = Plot title
  grid            = The bin size in (x,y) to be used
  minct           = Minimum count to be displayed
 OUTPUT::
  p               = Plot
 USAGE::
  p = zr_density_pdf(ax,Z,R,**args)
    """
# HISTORY::
#  10 Mar 2014 - Nick Guy NOAA/NSSL/WRDD, NRC
#--------------------------------------------------------
    # First calculate the 2D PDF
    pdf2_Z_R, xedges, yedges = gl.hist2d_masked(10.*np.log10(Z),np.log10(R),bins=grid,
                                                rng=(ylims,xlims),norm=True)
    
    # Replace any zero values with missing data for nice plots
    pdf2_Z_R = np.ma.masked_equal(pdf2_Z_R,0)
    
    # Flip and rotate the 2D Histogram
    pdf2_Z_R = np.rot90(pdf2_Z_R)
    pdf2_Z_R = np.flipud(pdf2_Z_R)
    
    # Create 2D arrays from xedges, yedges
    X, Y = np.meshgrid(xedges, yedges)
    
    # Plot the data using colormesh
    plt.pcolormesh(X,Y,pdf2_Z_R)#,cmap=

    plt.ylim(xlims) # Set X-axis limits
    plt.xlim(ylims) # Set Y-axis limits
    
    # First remove italics from mathtext then label axes
    plt.rc('mathtext',fontset='stixsans',default='regular') # Removes italics mathtext
    plt.xlabel(r'Reflectivity (dBZ)',labelpad=xpad,fontsize=ylabFontSize)
    plt.ylabel(r'log$_{10}$Rainfall Rate (mm h$^{-1}$)',labelpad=ypad,fontsize=xlabFontSize)
    
    ax.set_title(title) # Set the plot title
    
    # Add colorbar
    cb = plt.colorbar(shrink=0.85)
    cb.set_label('Normalized Counts')
    
    del pdf2_Z_R,xedges,yedges,X,Y
    
    return
Ejemplo n.º 2
0
def Nw_D0_density_pdf(ax,Nw,D0,title=None,grid=(36,61),minct=0.01,
                   xlabFontSize=16,xpad=7,ylabFontSize=16,ypad=7,
                   xlims=(0.1,3.5),ylims=(1.0,7.0)):
    """Create a density scatterplot using 2D histogram
 INPUT::
  ax              = Axis to plot against (produced externally)
  Nw              = Normalized intercept parameter [mm^-1 m^-3]
  D0              = Median volume diameter [mm]
     OPTIONAL
  title           = Plot title
  grid            = The bin size in (x,y) to be used
  minct           = Minimum count to be displayed
 OUTPUT::
  p               = Plot
 USAGE::
  p = Nw_D0_density_pdf(ax,Nw,D0,**args)
    """
# HISTORY::
#  10 Mar 2014 - Nick Guy NOAA/NSSL/WRDD, NRC
#--------------------------------------------------------
    # First calculate the 2D PDF
    pdf2_Nw_D0, xedges, yedges = gl.hist2d_masked(D0,np.log10(Nw),bins=grid,
                                                rng=(ylims,xlims),norm=True)
    
    # Replace any zero values with missing data for nice plots
    pdf2_Nw_D0 = np.ma.masked_equal(pdf2_Nw_D0,0)
    
    # Flip and rotate the 2D Histogram
    pdf2_Nw_D0 = np.rot90(pdf2_Nw_D0)
    pdf2_Nw_D0 = np.flipud(pdf2_Nw_D0)
    
    # Create 2D arrays from xedges, yedges
    X, Y = np.meshgrid(xedges, yedges)
    
    # Plot the data using colormesh
    plt.pcolormesh(X,Y,pdf2_Nw_D0)#,cmap=

    plt.ylim(xlims) # Set X-axis limits
    plt.xlim(ylims) # Set Y-axis limits
    
    # First remove italics from mathtext then label axes
    plt.rc('mathtext',fontset='stixsans',default='regular') # Removes italics mathtext
    plt.xlabel(r'D$_0$ (mm)',labelpad=xpad,fontsize=ylabFontSize)
    plt.ylabel(r'log$_{10}$[N$_w$] (mm$^{-1}$ m$^{-3}$)',labelpad=ypad,fontsize=xlabFontSize)
    
    ax.set_title(title) # Set the plot title
    
    # Add colorbar
    cb = plt.colorbar(shrink=0.85)
    cb.set_label('Normalized Counts')
    
    del pdf2_Nw_D0,xedges,yedges,X,Y
    
    return