plots.py 文件源码

def twohists(x1,x2,xmin,xmax,range=None,x1leg='$x_1$',x2leg='$x_2$',xlabel='',fig=1,sharey=False,fontsize=12,bins1=10,bins2=10):
    """
Script that plots two histograms of quantities x1 and x2
sharing the same X-axis.

:param x1,x2: arrays with data to be plotted
:param xmin,xmax: lower and upper range of plotted values, will be used to set a consistent x-range
    for both histograms.
:param x1leg, x2leg: legends for each histogram 
:param xlabel: self-explanatory.
:param bins1,bins2: number of bins in each histogram
:param fig: which plot window should I use?
:param range: in the form (xmin,xmax), same as range argument for hist and applied to both
    histograms.

Inspired by `Scipy <http://www.scipy.org/Cookbook/Matplotlib/Multiple_Subplots_with_One_Axis_Label>`_.
    """

    pylab.rcParams.update({'font.size': fontsize})
    fig=pylab.figure(fig)
    pylab.clf()

    a=fig.add_subplot(2,1,1)
    if sharey==True:
        b=fig.add_subplot(2,1,2, sharex=a, sharey=a)
    else:
        b=fig.add_subplot(2,1,2, sharex=a)

    a.hist(x1,bins1,label=x1leg,color='b',histtype='stepfilled',range=range)
    a.legend(loc='best',frameon=False)
    a.set_xlim(xmin,xmax)

    b.hist(x2,bins2,label=x2leg,color='r',histtype='stepfilled',range=range)
    b.legend(loc='best',frameon=False)

    pylab.setp(a.get_xticklabels(), visible=False)

    b.set_xlabel(xlabel)
    b.set_ylabel('Number',verticalalignment='bottom')
    pylab.minorticks_on()
    pylab.subplots_adjust(hspace=0.15)
    pylab.draw()
    pylab.show()