在matplotlib contourf图上绘制流线

Question

我有一张matplotlib等值线图，上面是经度和垂直方向的压强。我正在尝试使用matplotlib中的plt.streamplot函数并使用U和V风数据来绘制流线。如果我只绘制streamplot，它工作得很好。但是我不能把流线叠加到等高线上。以下是我的代码：

fig, axes = plt.subplots(nrows, ncols, sharex=True, sharey=True)
if (nrows==1 and ncols==1):
    axes=[axes]
else:
    axes=axes.flat
for i, ax in enumerate(axes):
    X,Y = np.meshgrid(x[i],y[i])
    levels=np.arange(vmin,vmax,step)
    h = ax.contourf(X,Y,z[i],cmap=cmap,levels=levels,extend='both')
    w = ax.streamplot(X, Y, W[i], Z[i], linewidth=0.2, color='gray')

这是我得到的图：

​

​

下面是流线图，不确定为什么y轴是从0到120而不是0到1000：

​

​

Answer 1

使用曲线坐标系绘制等高线(lat-p)。你必须将u，v转换为等高线的坐标系，就像这样(这是后面的例子，你必须修改它以使用压力水平)：

def myStreamPlot(lon,lat,u,v,color='k',density=2.5):
        from scipy.interpolate import griddata

        n,m = u.shape[1],u.shape[0]
        x = np.linspace(np.nanmin(lon), np.nanmax(lon), n)
        y = np.linspace(np.nanmin(lat), np.nanmax(lat), m)
        xi, yi = np.meshgrid(x,y)

        lon = lon.ravel()
        lat = lat.ravel()
        u   = u.ravel()
        v   = v.ravel()

        gu = griddata(zip(lon,lat), u, (xi,yi))
        gv = griddata(zip(lon,lat), v, (xi,yi))
        gspd = np.sqrt(gu**2 + gv**2)
        SL = plt.streamplot(x,y,gu,gv,linewidth=1.,color=color,density=density)

此代码使用scipy.interpolate的griddata函数：https://docs.scipy.org/doc/scipy/reference/generated/scipy.interpolate.griddata.html