求助风场流线图

wangweiqing117

最近在学习py中，遇到了一个不懂的问题，望大佬们给个帮助！
代码如下 ：
import netCDF4 as nc
import matplotlib.pyplot as plt
from mpl_toolkits.basemap import Basemap
import numpy as np
obj1=nc.Dataset('D:\\ST\\vwnd.sfc.2020.nc')
obj2=nc.Dataset('D:\\ST\\uwnd.sfc.2020.nc')

lat=obj1.variables['lat'][24:42]
lon=obj1.variables['lon'][24:52]
#x1, y1 =np.meshgrid(lon, lat)
u=obj2.variables['uwnd'][:,24:42,24:52]
v=obj1.variables['vwnd'][:,24:42,24:52]

for i in range(1):

    fig = plt.figure(figsize=(8, 6))
    m=Basemap(projection='stere',lon_0=90, lat_0=10,
              llcrnrlat=-10,llcrnrlon=60,
              urcrnrlat=30,urcrnrlon=127,resolution='i')

    speed = np.sqrt(u[i]**2 + v[i]**2)

    x, y = m.makegrid(v.shape[2], v.shape[1], returnxy=True)[2:8]

    m.streamplot(x, y, u[i], v[i], color=speed, cmap=plt.cm.winter, linewidth=0.2*speed)
    m.drawparallels(np.arange(0.,30.,10.),labels=[1,0,0,0],fontsize=15)
    m.drawmeridians(np.arange(60.,127.,15.),labels=[0,0,0,1],fontsize=15)
    m.readshapefile('d:\\ST\\country1','country1.shp',linewidth=1,drawbounds=True,color='gray')
    m.drawlsmask()


    plt.title('streamplot',size=20)

wangweiqing117

自己顶一下，不要沉了