2

我有一个垂直经度和压力水平的 matplotlib contourf 图。我正在尝试使用 matplotlib 中的 plt.streamplot 函数并使用 U 和 V 风数据来绘制流线。如果我只绘制流图，它工作正常。但我无法让流线覆盖在等高线图上。这是我的代码：-

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：

这是流线图，不知道为什么 y 轴是 0-120 而不是 0 到 1000

4

1 回答 1

1

您使用曲线坐标系绘制等高线图 (lat-p)。您必须将 u,v 转换为类似于此处的轮廓坐标系（这是 lat-lon 的示例，您必须修改它以使用压力水平）：

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)

此代码使用griddata功能scipy.interpolate：https ://docs.scipy.org/doc/scipy/reference/generated/scipy.interpolate.griddata.html

于 2019-02-14T04:40:26.003 回答