skewt.py 文件源码

def _windbarbs(u, v, press, delta):
    #delta = 2500 # equals 25mb
    p_bin_min = int((np.min(press) // delta) * delta)
    p_bin_max = int(((np.max(press) // delta)+1) * delta)
    p_bins = np.array(range(p_bin_min, p_bin_max, delta))

    ixs = np.digitize(press, p_bins)
    uwind = [np.mean(u[ixs == ix]) for ix in list(set(ixs))]
    vwind = [np.mean(v[ixs == ix]) for ix in list(set(ixs))]

    ax = plt.gca()
    inv = ax.transLimits.inverted()
    #x_pos, _none = inv.transform((0.92, 0))
    x_pos = inv.transform(np.array([[0.92,0]]))[0, 0]
    baraxis = [x_pos] * len(p_bins) 
    plt.barbs(baraxis, p_bins, uwind, vwind, \
              barb_increments=barb_increments, linewidth = .75)#, transform=ax.transAxes)