如何使用 matplotlib 和 pygrib 在 shapefile 上叠加降水数据?

How to overlay precipitation data on a shapefile using matplotlib and pygrib?

我读过这个问题 - Overlay shape file on matplotlib 我不确定它是否适用于我,但我有我所在国家/地区的 shapefile,我想将 GRIB 文件中的降水数据叠加到该 shapefile 上。这是使用 pygrib 和 matplotlib。我对使用 Basemap.coastlines() 和 Basemap.countries() 不感兴趣,因为它们在 shapefile 中可用。我想用我自己的 shapefile 覆盖它。我该怎么做?

#!/usr/bin/python3.4
import numpy as np
import pygrib
import shapefile
import matplotlib.pyplot as plt
from mpl_toolkits.basemap import Basemap


grbs = pygrib.open('00020000')

grb = grbs.select(name='Total Precipitation')[0]
data = grb.values
lat,lon = grb.latlons()
m=Basemap(projection='mill',lat_ts=10,llcrnrlon=lon.min(), \
urcrnrlon=lon.max(),llcrnrlat=lat.min(),urcrnrlat=lat.max(), \
resolution='c')
m.readshapefile("adm3.shp")
x,y=m(lon,lat)
cs = m.pcolormesh(x,y,data,shading='flat',cmap=plt.cm.jet)


m.drawparallels(np.arange(-90.,80.,5.),labels=[1,0,0,0])
m.drawmeridians(np.arange(-180.,180.,5.),labels=[0,0,0,1])
plt.colorbar(cs,orientation='vertical')
plt.title('Precipitation Plot')
plt.show()

我不确定我是否理解,因为你的代码似乎基本上可以工作。我做了一些改动:

  • resolution kwarg in Basemap 可以设置为 None 因为你不会使用内置的海岸线
  • m.readshapefile 需要一个 name 参数

正在使用

import numpy as np
import matplotlib.pyplot as plt
from mpl_toolkits.basemap import Basemap

grbs = pygrib.open('00020000.grb')
grb = grbs.select(name='Total Precipitation')[0]
data = grb.values
lat, lon = grb.latlons()

m = Basemap(projection='mill', lat_ts=10,
            llcrnrlon=lon.min(),
            urcrnrlon=lon.max(),
            llcrnrlat=lat.min(),
            urcrnrlat=lat.max(),
            resolution=None)

m.readshapefile("adm3", "countryname", color="red", linewidth=3)
x, y = m(lon,lat)
cs = m.pcolormesh(x,y,data,shading='flat',cmap=plt.cm.jet)

m.drawparallels(np.arange(-90.,80.,5.),labels=[1,0,0,0])
m.drawmeridians(np.arange(-180.,180.,5.),labels=[0,0,0,1])
plt.colorbar(cs,orientation='vertical')
plt.title('Precipitation Plot')
plt.show()

和一些模拟数据和国家边界,我得到如下图: