def draw_cumulated_precip_evo(
rain=None,
map_extent=(50, 150, 0, 65),
regrid_shape=20,
add_china=True,city=True,south_China_sea=True,
output_dir=None,Global=False):
# set font
plt.rcParams['font.sans-serif'] = ['SimHei'] # 步骤一（替换sans-serif字体）
plt.rcParams['axes.unicode_minus'] = False # 步骤二（解决坐标轴负数的负号显示问题）
# set figure
fig = plt.figure(figsize=(16,9))
plotcrs = ccrs.AlbersEqualArea(central_latitude=(map_extent[2]+map_extent[3])/2.,
central_longitude=(map_extent[0]+map_extent[1])/2., standard_parallels=[30., 60.])
datacrs = ccrs.PlateCarree()
ax = plt.axes([0.01,0.1,.98,.84], projection=plotcrs)
map_extent2=utl.adjust_map_ratio(ax,map_extent=map_extent,datacrs=datacrs)
# define return plots
plots = {}
# draw mean sea level pressure
if rain is not None:
x, y = np.meshgrid(rain['lon'], rain['lat'])
z=np.squeeze(rain['data'].values)
z[z<0.1]=np.nan
znan=np.zeros(shape=x.shape)
znan[:]=np.nan
cmap,norm=dk_ctables.cm_qpf_nws(atime=24)
cmap.set_under(color=[0,0,0,0],alpha=0.0)
plots['rain'] = ax.pcolormesh(
x,y,znan, norm=norm,
cmap=cmap, zorder=1,transform=datacrs,alpha=0.5)
#additional information
plt.title('['+rain.attrs['model']+'] '+
str(int(rain.coords['forecast_period'].values[0])-rain.attrs['t_gap'])+
'至'+str(int(rain.coords['forecast_period'].values[-1]))+'时效累积降水预报',
loc='left', fontsize=25)
ax.add_feature(cfeature.OCEAN)
utl.add_china_map_2cartopy_public(
ax, name='coastline', edgecolor='gray', lw=0.8, zorder=3,alpha=0.5)
if add_china:
utl.add_china_map_2cartopy_public(
ax, name='province', edgecolor='gray', lw=0.5, zorder=3)
utl.add_china_map_2cartopy_public(
ax, name='nation', edgecolor='black', lw=0.8, zorder=3)
utl.add_china_map_2cartopy_public(
ax, name='river', edgecolor='#74b9ff', lw=0.8, zorder=3,alpha=0.5)
# grid lines
gl = ax.gridlines(
crs=datacrs, linewidth=2, color='gray', alpha=0.5, linestyle='--', zorder=1)
gl.xlocator = mpl.ticker.FixedLocator(np.arange(0, 360, 15))
gl.ylocator = mpl.ticker.FixedLocator(np.arange(-90, 90, 15))
utl.add_cartopy_background(ax,name='RD')
l, b, w, h = ax.get_position().bounds
#forecast information
initTime = pd.to_datetime(
str(rain.coords['forecast_reference_time'].values)).replace(tzinfo=None).to_pydatetime()
#发布时间
if(sys.platform[0:3] == 'lin'):
locale.setlocale(locale.LC_CTYPE, 'zh_CN.utf8')
if(sys.platform[0:3] == 'win'):
locale.setlocale(locale.LC_CTYPE, 'chinese')
# add color bar
cax=plt.axes([l,b-0.04,w,.02])
cb = plt.colorbar(plots['rain'], cax=cax, orientation='horizontal')
cb.ax.tick_params(labelsize='x-large')
cb.set_label('Precipitation (mm)',size=20)
fcst_time=initTime+timedelta(hours=rain.coords['forecast_period'].values[0])
bax=plt.axes([l,b+h-0.1,.25,.1])
bax.set_yticks([])
bax.set_xticks([])
bax.axis([0, 10, 0, 10])
fcst_time1=initTime+timedelta(hours=rain.coords['forecast_period'].values[0]-6)
bax.text(2.5, 7.5,'起始时间: '+fcst_time1.strftime("%Y年%m月%d日%H时"),size=15)
bax.text(2.5, 0.5,'www.nmc.cn',size=15)
valid_fhour=bax.text(2.5, 5,'截至时间: ',size=15)
txt_fhour=bax.text(2.5, 2.5,'预报时效: ',size=15)
utl.add_logo_extra_in_axes(pos=[l-0.02,b+h-0.1,.1,.1],which='nmc', size='Xlarge')
# add south China sea
if south_China_sea:
utl.add_south_China_sea(pos=[l+w-0.091,b,.1,.2])
small_city=False
if(map_extent2[1]-map_extent2[0] < 25):
small_city=True
if city:
utl.add_city_on_map(ax,map_extent=map_extent2,transform=datacrs,zorder=2,size=13,small_city=small_city)
def update(frame_number):
fcst_time2=initTime+timedelta(hours=rain.coords['forecast_period'].values[frame_number])
valid_fhour.set_text('截至时间: '+fcst_time2.strftime("%Y年%m月%d日%H时"))
txt_fhour.set_text('预报时效: '+str(int(rain.coords['forecast_period'].values[frame_number]))+'小时')
return ax.pcolormesh(
x,y,np.squeeze(z[frame_number,:,:]), norm=norm,
cmap=cmap, zorder=1,transform=datacrs,alpha=0.5)
nframes=rain['data'].shape[0]
animation1 = FuncAnimation(fig, update, frames=nframes,interval=1000)
# ffmpegpath = os.path.abspath(r"C:\Users\HEYGY\Desktop\ffmpeg-20200824-3477feb-win64-static\bin\ffmpeg.exe")
# import matplotlib
# matplotlib.rcParams["animation.ffmpeg_path"] = ffmpegpath
# writer = animation.FFMpegWriter()
# show figure
plt.subplots_adjust(top=1, bottom=0, right=0.93, left=0, hspace=0, wspace=0)
plt.margins(0, 0)
if(output_dir != None):
animation1.save(output_dir+'累积降水演变_'+
'起报时间_'+initTime.strftime("%Y年%m月%d日%H时")+
'预报时效_'+str(int(rain.coords['forecast_period'].values[-1]))+'小时_'+
'['+rain.attrs['model']+'] '+'.gif',writer='pillow')
if(output_dir == None):
#animation.save('rain.gif', fps=75, writer='imagemagick')
plt.show()
def draw_gh_rain(gh=None, rain=None,
map_extent=(50, 150, 0, 65),
regrid_shape=20,
add_china=True,city=True,south_China_sea=True,
output_dir=None,Global=False):
# set font
plt.rcParams['font.sans-serif'] = ['SimHei'] # 步骤一（替换sans-serif字体）
plt.rcParams['axes.unicode_minus'] = False # 步骤二（解决坐标轴负数的负号显示问题）
# set figure
plt.figure(figsize=(16,9))
if(Global == True):
plotcrs = ccrs.Robinson(central_longitude=115.)
else:
plotcrs = ccrs.AlbersEqualArea(central_latitude=(map_extent[2]+map_extent[3])/2.,
central_longitude=(map_extent[0]+map_extent[1])/2., standard_parallels=[30., 60.])
datacrs = ccrs.PlateCarree()
ax = plt.axes([0.01,0.1,.98,.84], projection=plotcrs)
map_extent2=utl.adjust_map_ratio(ax,map_extent=map_extent,datacrs=datacrs)
# define return plots
plots = {}
# draw mean sea level pressure
if rain is not None:
x, y = np.meshgrid(rain['lon'], rain['lat'])
z=np.squeeze(rain['data'].values)
z[z<0.1]=np.nan
cmap,norm=dk_ctables.cm_qpf_nws(atime=rain.attrs['atime'])
cmap.set_under(color=[0,0,0,0],alpha=0.0)
plots['rain'] = ax.pcolormesh(
x,y,z, norm=norm,
cmap=cmap, zorder=1,transform=datacrs,alpha=0.5)
# draw -hPa geopotential height
if gh is not None:
x, y = np.meshgrid(gh['lon'], gh['lat'])
clevs_gh = np.append(np.append(np.arange(0, 480, 4),np.append(np.arange(480, 584, 8), np.arange(580, 604, 4))), np.arange(604, 2000, 8))
plots['gh'] = ax.contour(
x, y, np.squeeze(gh['data']), clevs_gh, colors='black',
linewidths=2, transform=datacrs, zorder=3)
plt.clabel(plots['gh'], inline=2, fontsize=20, fmt='%.0f',colors='black')
#additional information
plt.title('['+gh.attrs['model']+'] '+
str(int(gh['level'].values[0]))+'hPa 位势高度场, '+
str(int(rain.attrs['atime']))+'小时降水',
loc='left', fontsize=30)
ax.add_feature(cfeature.OCEAN)
utl.add_china_map_2cartopy_public(
ax, name='coastline', edgecolor='gray', lw=0.8, zorder=3,alpha=0.5)
if add_china:
utl.add_china_map_2cartopy_public(
ax, name='province', edgecolor='gray', lw=0.5, zorder=3)
utl.add_china_map_2cartopy_public(
ax, name='nation', edgecolor='black', lw=0.8, zorder=3)
utl.add_china_map_2cartopy_public(
ax, name='river', edgecolor='#74b9ff', lw=0.8, zorder=3,alpha=0.5)
# grid lines
gl = ax.gridlines(
crs=datacrs, linewidth=2, color='gray', alpha=0.5, linestyle='--', zorder=1)
gl.xlocator = mpl.ticker.FixedLocator(np.arange(0, 360, 15))
gl.ylocator = mpl.ticker.FixedLocator(np.arange(-90, 90, 15))
utl.add_cartopy_background(ax,name='RD')
l, b, w, h = ax.get_position().bounds
#forecast information
bax=plt.axes([l,b+h-0.1,.25,.1],facecolor='#FFFFFFCC')
bax.set_yticks([])
bax.set_xticks([])
bax.axis([0, 10, 0, 10])
initTime = pd.to_datetime(
str(gh.coords['forecast_reference_time'].values)).replace(tzinfo=None).to_pydatetime()
fcst_time=initTime+timedelta(hours=gh.coords['forecast_period'].values[0])
#发布时间
if(sys.platform[0:3] == 'lin'):
locale.setlocale(locale.LC_CTYPE, 'zh_CN.utf8')
if(sys.platform[0:3] == 'win'):
locale.setlocale(locale.LC_CTYPE, 'chinese')
plt.text(2.5, 7.5,'起报时间: '+initTime.strftime("%Y年%m月%d日%H时"),size=15)
plt.text(2.5, 5,'预报时间: '+fcst_time.strftime("%Y年%m月%d日%H时"),size=15)
plt.text(2.5, 2.5,'预报时效: '+str(int(gh.coords['forecast_period'].values[0]))+'小时'+'(降水'+str(int(gh.coords['forecast_period'].values[0]+12))+'小时)',size=15)
plt.text(2.5, 0.5,'www.nmc.cn',size=15)
# add color bar
if(rain != None):
cax=plt.axes([l,b-0.04,w,.02])
cb = plt.colorbar(plots['rain'], cax=cax, orientation='horizontal')
cb.ax.tick_params(labelsize='x-large')
cb.set_label(str(int(rain.attrs['atime']))+'h precipitation (mm)',size=20)
# add south China sea
if south_China_sea:
utl.add_south_China_sea(pos=[l+w-0.091,b,.1,.2])
small_city=False
if(map_extent2[1]-map_extent2[0] < 25):
small_city=True
if city:
utl.add_city_on_map(ax,map_extent=map_extent2,transform=datacrs,zorder=2,size=13,small_city=small_city)
utl.add_logo_extra_in_axes(pos=[l-0.02,b+h-0.1,.1,.1],which='nmc', size='Xlarge')
# show figure
if(output_dir != None):
plt.savefig(output_dir+'高度场_降水_预报_'+
'起报时间_'+initTime.strftime("%Y年%m月%d日%H时")+
'预报时效_'+str(int(gh.coords['forecast_period'].values[0]))+'小时'+'.png', dpi=200)
if(output_dir == None):
plt.show()
def draw_Rain_evo(
rain=None,fcs_lvl=4,
map_extent=(50, 150, 0, 65),
regrid_shape=20,
add_china=True,city=True,south_China_sea=True,
output_dir=None,Global=False):
# set font
plt.rcParams['font.sans-serif'] = ['SimHei'] # 步骤一（替换sans-serif字体）
plt.rcParams['axes.unicode_minus'] = False # 步骤二（解决坐标轴负数的负号显示问题）
if(sys.platform[0:3] == 'lin'):
locale.setlocale(locale.LC_CTYPE, 'zh_CN.utf8')
if(sys.platform[0:3] == 'win'):
locale.setlocale(locale.LC_CTYPE, 'chinese')
# set figure
plt.figure(figsize=(16,9))
if(Global == True):
plotcrs = ccrs.Robinson(central_longitude=115.)
else:
plotcrs = ccrs.AlbersEqualArea(central_latitude=(map_extent[2]+map_extent[3])/2.,
central_longitude=(map_extent[0]+map_extent[1])/2., standard_parallels=[30., 60.])
ax = plt.axes([0.01,0.1,.98,.84], projection=plotcrs)
datacrs = ccrs.PlateCarree()
map_extent2=utl.adjust_map_ratio(ax,map_extent=map_extent,datacrs=datacrs)
plt.title('['+rain.attrs['model']+'] 预报逐'+
str(rain.attrs['t_gap'])+'小时'+str(fcs_lvl)+'mm降水范围演变',
loc='left', fontsize=30)
#draw data
plots = {}
if rain is not None:
x, y = np.meshgrid(rain['lon'], rain['lat'])
for itime in range(0,len(rain['time'].values)):
z=np.squeeze(rain['data'].values[itime,:,:])
z[z<=0.1]=np.nan
initTime = pd.to_datetime(str(rain.coords['forecast_reference_time'].values)).replace(tzinfo=None).to_pydatetime()
labels=(initTime+timedelta(hours=rain.coords['forecast_period'].values[itime])).strftime("%m月%d日%H时")
cmap=mpl.cm.jet
per_color=utl.get_part_clev_and_cmap(cmap=cmap,clev_range=[0,len(rain['time'].values)],clev_slt=itime)
ax.contourf(
x,y,z, levels=[fcs_lvl,800],
colors=per_color, zorder=3,transform=datacrs,
alpha=0.2+itime*((1-0.2)/len(rain['time'].values)))
if(itime == 0):
label_handles = [mpatches.Patch(color=per_color.reshape(4),
alpha=0.2+itime*((1-0.2)/len(rain['time'].values)), label=labels)]
else:
label_handles.append(mpatches.Patch(color=per_color.reshape(4),alpha=0.2+itime*((1-0.2)/len(rain['time'].values)), label=labels))
leg = plt.legend(handles=label_handles, loc=3,framealpha=1)
#additional information
ax.add_feature(cfeature.OCEAN)
utl.add_china_map_2cartopy_public(
ax, name='coastline', edgecolor='gray', lw=0.8, zorder=1,alpha=0.5)
if add_china:
utl.add_china_map_2cartopy_public(
ax, name='province', edgecolor='gray', lw=0.5, zorder=1)
utl.add_china_map_2cartopy_public(
ax, name='nation', edgecolor='black', lw=0.8, zorder=1)
utl.add_china_map_2cartopy_public(
ax, name='river', edgecolor='#74b9ff', lw=0.8, zorder=1,alpha=0.5)
# grid lines
gl = ax.gridlines(
crs=datacrs, linewidth=2, color='gray', alpha=0.5, linestyle='--', zorder=1)
gl.xlocator = mpl.ticker.FixedLocator(np.arange(0, 360, 15))
gl.ylocator = mpl.ticker.FixedLocator(np.arange(-90, 90, 15))
utl.add_cartopy_background(ax,name='RD')
l, b, w, h = ax.get_position().bounds
#forecast information
bax=plt.axes([l,b+h-0.1,.25,.1],facecolor='#FFFFFFCC')
bax.set_yticks([])
bax.set_xticks([])
bax.axis([0, 10, 0, 10])
initTime = pd.to_datetime(
str(rain.coords['forecast_reference_time'].values)).replace(tzinfo=None).to_pydatetime()
fcst_time=initTime+timedelta(hours=rain.coords['forecast_period'].values[0])
#发布时间
if(sys.platform[0:3] == 'lin'):
locale.setlocale(locale.LC_CTYPE, 'zh_CN.utf8')
if(sys.platform[0:3] == 'win'):
locale.setlocale(locale.LC_CTYPE, 'chinese')
plt.text(2.5, 7.5,'起报时间: '+initTime.strftime("%Y年%m月%d日%H时"),size=15)
plt.text(2.5, 5,'起始时间: '+
(initTime+timedelta(hours=rain.coords['forecast_period'].values[0])).strftime("%Y年%m月%d日%H时"),size=15)
plt.text(2.5, 2.5,'终止时间: '+
(initTime+timedelta(hours=rain.coords['forecast_period'].values[1])).strftime("%Y年%m月%d日%H时"),size=15)
plt.text(2.5, 0.5,'www.nmc.cn',size=15)
# add color bar
# add south China sea
if south_China_sea:
utl.add_south_China_sea(pos=[l+w-0.091,b,.1,.2])
small_city=False
if(map_extent2[1]-map_extent2[0] < 25):
small_city=True
if city:
utl.add_city_on_map(ax,map_extent=map_extent2,transform=datacrs,zorder=110,size=13,small_city=small_city)
utl.add_logo_extra_in_axes(pos=[l-0.02,b+h-0.1,.1,.1],which='nmc', size='Xlarge')
# show figure
if(output_dir != None):
plt.savefig(output_dir+'海平面气压_降水_预报_'+
'起报时间_'+initTime.strftime("%Y年%m月%d日%H时")+
'预报时效_'+str(int(rain.coords['forecast_period'].values[0]))+'小时'+'.png', dpi=200)
if(output_dir == None):
plt.show()
def draw_gh_uv_mslp(gh=None,
uv=None,
mslp=None,
map_extent=(50, 150, 0, 65),
regrid_shape=20,
add_china=True,
city=True,
south_China_sea=True,
output_dir=None,
Global=False):
plt.rcParams['font.sans-serif'] = ['SimHei'] # 步骤一（替换sans-serif字体）
plt.rcParams['axes.unicode_minus'] = False # 步骤二（解决坐标轴负数的负号显示问题）
# draw figure
plt.figure(figsize=(16, 9))
# set data projection
if (Global == True):
plotcrs = ccrs.Robinson(central_longitude=115.)
else:
plotcrs = ccrs.AlbersEqualArea(
central_latitude=(map_extent[2] + map_extent[3]) / 2.,
central_longitude=(map_extent[0] + map_extent[1]) / 2.,
standard_parallels=[30., 60.])
ax = plt.axes([0.01, 0.1, .98, .84], projection=plotcrs)
plt.title('[' + gh.attrs['model'] + '] ' +
str(int(gh['level'].values[0])) + 'hPa 位势高度场, ' +
str(int(uv['level'].values[0])) + 'hPa 风场, 海平面气压场',
loc='left',
fontsize=30)
datacrs = ccrs.PlateCarree()
#adapt to the map ratio
map_extent2 = utl.adjust_map_ratio(ax,
map_extent=map_extent,
datacrs=datacrs)
#adapt to the map ratio
ax.add_feature(cfeature.OCEAN)
utl.add_china_map_2cartopy_public(ax,
name='coastline',
edgecolor='gray',
lw=0.8,
zorder=5,
alpha=0.5)
if add_china:
utl.add_china_map_2cartopy_public(ax,
name='province',
edgecolor='gray',
lw=0.5,
zorder=5)
utl.add_china_map_2cartopy_public(ax,
name='nation',
edgecolor='black',
lw=0.8,
zorder=5)
utl.add_china_map_2cartopy_public(ax,
name='river',
edgecolor='#74b9ff',
lw=0.8,
zorder=5,
alpha=0.5)
# define return plots
plots = {}
# draw mean sea level pressure
if mslp is not None:
x, y = np.meshgrid(mslp['lon'], mslp['lat'])
clevs_mslp = np.arange(960, 1065, 5)
cmap = guide_cmaps(26)
plots['mslp'] = ax.contourf(x,
y,
np.squeeze(mslp['data']),
clevs_mslp,
cmap=cmap,
alpha=0.8,
zorder=1,
transform=datacrs)
#+画高低压中心
res = mslp['lon'].values[1] - mslp['lon'].values[0]
nwindow = int(9.5 / res)
mslp_hl = np.ma.masked_invalid(mslp['data'].values).squeeze()
local_min, local_max = utl.extrema(mslp_hl,
mode='wrap',
window=nwindow)
#Get location of extrema on grid
xmin, xmax, ymin, ymax = map_extent2
lons2d, lats2d = x, y
transformed = datacrs.transform_points(datacrs, lons2d, lats2d)
x = transformed[..., 0]
y = transformed[..., 1]
xlows = x[local_min]
xhighs = x[local_max]
ylows = y[local_min]
yhighs = y[local_max]
lowvals = mslp_hl[local_min]
highvals = mslp_hl[local_max]
yoffset = 0.022 * (ymax - ymin)
dmin = yoffset
#Plot low pressures
xyplotted = []
for x, y, p in zip(xlows, ylows, lowvals):
if x < xmax - yoffset and x > xmin + yoffset and y < ymax - yoffset and y > ymin + yoffset:
dist = [
np.sqrt((x - x0)**2 + (y - y0)**2) for x0, y0 in xyplotted
]
if not dist or min(dist) > dmin: #,fontweight='bold'
a = ax.text(x,
y,
'L',
fontsize=28,
ha='center',
va='center',
color='r',
fontweight='normal',
transform=datacrs)
b = ax.text(x,
y - yoffset,
repr(int(p)),
fontsize=14,
ha='center',
va='top',
color='r',
fontweight='normal',
transform=datacrs)
a.set_path_effects([
path_effects.Stroke(linewidth=1.5, foreground='black'),
path_effects.SimpleLineShadow(),
path_effects.Normal()
])
b.set_path_effects([
path_effects.Stroke(linewidth=1.0, foreground='black'),
path_effects.SimpleLineShadow(),
path_effects.Normal()
])
xyplotted.append((x, y))
#Plot high pressures
xyplotted = []
for x, y, p in zip(xhighs, yhighs, highvals):
if x < xmax - yoffset and x > xmin + yoffset and y < ymax - yoffset and y > ymin + yoffset:
dist = [
np.sqrt((x - x0)**2 + (y - y0)**2) for x0, y0 in xyplotted
]
if not dist or min(dist) > dmin:
a = ax.text(x,
y,
'H',
fontsize=28,
ha='center',
va='center',
color='b',
fontweight='normal',
transform=datacrs)
b = ax.text(x,
y - yoffset,
repr(int(p)),
fontsize=14,
ha='center',
va='top',
color='b',
fontweight='normal',
transform=datacrs)
a.set_path_effects([
path_effects.Stroke(linewidth=1.5, foreground='black'),
path_effects.SimpleLineShadow(),
path_effects.Normal()
])
b.set_path_effects([
path_effects.Stroke(linewidth=1.0, foreground='black'),
path_effects.SimpleLineShadow(),
path_effects.Normal()
])
xyplotted.append((x, y))
#-画高低压中心
# draw -hPa wind bards
if uv is not None:
x, y = np.meshgrid(uv['lon'], uv['lat'])
u = np.squeeze(uv['u']) * 2.5
v = np.squeeze(uv['v']) * 2.5
plots['uv'] = ax.barbs(x,
y,
u.values,
v.values,
length=6,
regrid_shape=regrid_shape,
transform=datacrs,
fill_empty=False,
sizes=dict(emptybarb=0.05),
zorder=2)
# draw -hPa geopotential height
if gh is not None:
x, y = np.meshgrid(gh['lon'], gh['lat'])
clevs_gh = np.append(
np.append(
np.arange(0, 480, 4),
np.append(np.arange(480, 584, 8), np.arange(580, 604, 4))),
np.arange(604, 2000, 8))
linewidths_gh = np.zeros(clevs_gh.shape) + 2
idx_588 = np.where(clevs_gh == 588)
linewidths_gh[idx_588[0]] = 4
plots['gh'] = ax.contour(x,
y,
np.squeeze(gh['data']),
clevs_gh,
colors='purple',
linewidths=linewidths_gh,
transform=datacrs,
zorder=3)
ax.clabel(plots['gh'],
inline=1,
fontsize=20,
fmt='%.0f',
colors='black')
# grid lines
gl = ax.gridlines(crs=datacrs,
linewidth=2,
color='gray',
alpha=0.5,
linestyle='--',
zorder=4)
gl.xlocator = mpl.ticker.FixedLocator(np.arange(0, 360, 15))
gl.ylocator = mpl.ticker.FixedLocator(np.arange(-90, 90, 15))
utl.add_cartopy_background(ax, name='RD')
#forecast information
l, b, w, h = ax.get_position().bounds
bax = plt.axes([l, b + h - 0.1, .25, .1], facecolor='#FFFFFFCC')
bax.set_yticks([])
bax.set_xticks([])
bax.axis([0, 10, 0, 10])
initTime = pd.to_datetime(str(
gh.coords['forecast_reference_time'].values)).replace(
tzinfo=None).to_pydatetime()
fcst_time = initTime + timedelta(
hours=gh.coords['forecast_period'].values[0])
#发布时间
if (sys.platform[0:3] == 'lin'):
locale.setlocale(locale.LC_CTYPE, 'zh_CN.utf8')
if (sys.platform[0:3] == 'win'):
locale.setlocale(locale.LC_CTYPE, 'chinese')
plt.text(2.5, 7.5, '起报时间: ' + initTime.strftime("%Y年%m月%d日%H时"), size=15)
plt.text(2.5, 5, '预报时间: ' + fcst_time.strftime("%Y年%m月%d日%H时"), size=15)
plt.text(2.5,
2.5,
'预报时效: ' + str(int(gh.coords['forecast_period'].values[0])) +
'小时',
size=15)
plt.text(2.5, 0.5, 'www.nmc.cn', size=15)
# add color bar
cax = plt.axes([l, b - 0.04, w, .02])
cb = plt.colorbar(plots['mslp'],
cax=cax,
orientation='horizontal',
ticks=clevs_mslp[:-1],
extend='max',
extendrect=False)
cb.ax.tick_params(labelsize='x-large')
cb.set_label('Mean sea level pressure (hPa)', size=20)
# add south China sea
if south_China_sea:
utl.add_south_China_sea(pos=[l + w - 0.091, b, .1, .2])
small_city = False
if (map_extent2[1] - map_extent2[0] < 25):
small_city = True
if city:
utl.add_city_on_map(ax,
map_extent=map_extent2,
transform=datacrs,
zorder=6,
size=15,
small_city=small_city)
utl.add_logo_extra_in_axes(pos=[l - 0.02, b + h - 0.1, .1, .1],
which='nmc',
size='Xlarge')
# show figure
if (output_dir != None):
plt.savefig(output_dir + '最高温度_预报_' + '起报时间_' +
initTime.strftime("%Y年%m月%d日%H时") + '预报时效_' +
str(gh.coords['forecast_period'].values[0]) + '小时' +
'.png',
dpi=200,
bbox_inches='tight')
if (output_dir == None):
plt.show()
def draw_Miller_Composite_Chart(fcst_info=None,
u_300=None,
v_300=None,
u_500=None,
v_500=None,
u_850=None,
v_850=None,
pmsl_change=None,
hgt_500_change=None,
Td_dep_700=None,
Td_sfc=None,
pmsl=None,
lifted_index=None,
vort_adv_500_smooth=None,
map_extent=(50, 150, 0, 65),
add_china=True,
city=True,
south_China_sea=True,
output_dir=None,
Global=False):
# set font
plt.rcParams['font.sans-serif'] = ['SimHei'] # 步骤一（替换sans-serif字体）
plt.rcParams['axes.unicode_minus'] = False # 步骤二（解决坐标轴负数的负号显示问题）
# set figure
plt.figure(figsize=(16, 9))
if (Global == True):
plotcrs = ccrs.Robinson(central_longitude=115.)
else:
plotcrs = ccrs.AlbersEqualArea(
central_latitude=(map_extent[2] + map_extent[3]) / 2.,
central_longitude=(map_extent[0] + map_extent[1]) / 2.,
standard_parallels=[30., 60.])
datacrs = ccrs.PlateCarree()
ax = plt.axes([0.01, 0.1, .98, .84], projection=plotcrs)
map_extent2 = utl.adjust_map_ratio(ax,
map_extent=map_extent,
datacrs=datacrs)
#draw data
plots = {}
lons = fcst_info['lon']
lats = fcst_info['lat']
cs1 = ax.contour(lons,
lats,
lifted_index,
range(-8, -2, 2),
transform=ccrs.PlateCarree(),
colors='red',
linewidths=0.75,
linestyles='solid',
zorder=7)
cs1.clabel(fontsize=10,
inline=1,
inline_spacing=7,
fmt='%i',
rightside_up=True,
use_clabeltext=True)
# Plot Surface pressure falls
cs2 = ax.contour(lons,
lats,
pmsl_change.to('hPa'),
range(-10, -1, 4),
transform=ccrs.PlateCarree(),
colors='k',
linewidths=0.75,
linestyles='dashed',
zorder=6)
cs2.clabel(fontsize=10,
inline=1,
inline_spacing=7,
fmt='%i',
rightside_up=True,
use_clabeltext=True)
# Plot 500-hPa height falls
cs3 = ax.contour(lons,
lats,
hgt_500_change,
range(-60, -29, 15),
transform=ccrs.PlateCarree(),
colors='k',
linewidths=0.75,
linestyles='solid',
zorder=5)
cs3.clabel(fontsize=10,
inline=1,
inline_spacing=7,
fmt='%i',
rightside_up=True,
use_clabeltext=True)
# Plot surface pressure
ax.contourf(lons,
lats,
pmsl.to('hPa'),
range(990, 1011, 20),
alpha=0.5,
transform=ccrs.PlateCarree(),
colors='yellow',
zorder=1)
# Plot surface dewpoint
ax.contourf(lons,
lats,
Td_sfc.to('degF'),
range(65, 76, 10),
alpha=0.4,
transform=ccrs.PlateCarree(),
colors=['green'],
zorder=2)
# Plot 700-hPa dewpoint depression
ax.contourf(lons,
lats,
Td_dep_700,
range(15, 46, 30),
alpha=0.5,
transform=ccrs.PlateCarree(),
colors='tan',
zorder=3)
# Plot Vorticity Advection
ax.contourf(lons,
lats,
vort_adv_500_smooth,
range(5, 106, 100),
alpha=0.5,
transform=ccrs.PlateCarree(),
colors='BlueViolet',
zorder=4)
# Define a skip to reduce the barb point density
skip_300 = (slice(None, None, 12), slice(None, None, 12))
skip_500 = (slice(None, None, 10), slice(None, None, 10))
skip_850 = (slice(None, None, 8), slice(None, None, 8))
x, y = np.meshgrid(fcst_info['lon'], fcst_info['lat'])
# 300-hPa wind barbs
jet300 = ax.barbs(x[skip_300],
y[skip_300],
u_300[skip_300].m,
v_300[skip_300].m,
length=6,
transform=ccrs.PlateCarree(),
color='green',
zorder=10,
label='300-hPa Jet Core Winds (m/s)')
# 500-hPa wind barbs
jet500 = ax.barbs(x[skip_500],
y[skip_500],
u_500[skip_500].m,
v_500[skip_500].m,
length=6,
transform=ccrs.PlateCarree(),
color='blue',
zorder=9,
label='500-hPa Jet Core Winds (m/s)')
# 850-hPa wind barbs
jet850 = ax.barbs(x[skip_850],
y[skip_850],
u_850[skip_850].m,
v_850[skip_850].m,
length=6,
transform=ccrs.PlateCarree(),
color='k',
zorder=8,
label='850-hPa Jet Core Winds (m/s)')
#additional information
plt.title('[' + fcst_info['model'] + '] ' + 'Miller 综合分析图',
loc='left',
fontsize=30)
ax.add_feature(cfeature.OCEAN)
utl.add_china_map_2cartopy_public(ax,
name='coastline',
edgecolor='gray',
lw=0.8,
zorder=105,
alpha=0.5)
if add_china:
utl.add_china_map_2cartopy_public(ax,
name='province',
edgecolor='gray',
lw=0.5,
zorder=105)
utl.add_china_map_2cartopy_public(ax,
name='nation',
edgecolor='black',
lw=0.8,
zorder=105)
utl.add_china_map_2cartopy_public(ax,
name='river',
edgecolor='#74b9ff',
lw=0.8,
zorder=105,
alpha=0.5)
gl = ax.gridlines(crs=datacrs,
linewidth=2,
color='gray',
alpha=0.5,
linestyle='--',
zorder=40)
gl.xlocator = mpl.ticker.FixedLocator(np.arange(0, 360, 15))
gl.ylocator = mpl.ticker.FixedLocator(np.arange(-90, 90, 15))
utl.add_cartopy_background(ax, name='RD')
l, b, w, h = ax.get_position().bounds
# Legend
purple = mpatches.Patch(color='BlueViolet',
label='Cyclonic Absolute Vorticity Advection')
yellow = mpatches.Patch(color='yellow', label='Surface MSLP < 1010 hPa')
green = mpatches.Patch(color='green', label='Surface Td > 65 F')
tan = mpatches.Patch(color='tan',
label='700 hPa Dewpoint Depression > 15 C')
red_line = lines.Line2D([], [], color='red', label='Best Lifted Index (C)')
dashed_black_line = lines.Line2D(
[], [],
linestyle='dashed',
color='k',
label='12-hr Surface Pressure Falls (hPa)')
black_line = lines.Line2D([], [],
linestyle='solid',
color='k',
label='12-hr 500-hPa Height Falls (m)')
leg = plt.legend(handles=[
jet300, jet500, jet850, dashed_black_line, black_line, red_line,
purple, tan, green, yellow
],
loc=3,
title='Composite Analysis Valid: ',
framealpha=1)
leg.set_zorder(100)
#forecast information
bax = plt.axes([l, b + h - 0.1, .25, .1], facecolor='#FFFFFFCC')
bax.set_yticks([])
bax.set_xticks([])
bax.axis([0, 10, 0, 10])
initTime = pd.to_datetime(str(
fcst_info['forecast_reference_time'])).replace(
tzinfo=None).to_pydatetime()
fcst_time = initTime + timedelta(hours=fcst_info['forecast_period'])
#发布时间
if (sys.platform[0:3] == 'lin'):
locale.setlocale(locale.LC_CTYPE, 'zh_CN.utf8')
if (sys.platform[0:3] == 'win'):
locale.setlocale(locale.LC_CTYPE, 'chinese')
plt.text(2.5, 7.5, '起报时间: ' + initTime.strftime("%Y年%m月%d日%H时"), size=15)
plt.text(2.5, 5, '预报时间: ' + fcst_time.strftime("%Y年%m月%d日%H时"), size=15)
plt.text(2.5,
2.5,
'预报时效: ' + str(fcst_info['forecast_period']) + '小时',
size=15)
plt.text(2.5, 0.5, 'www.nmc.cn', size=15)
# add south China sea
if south_China_sea:
utl.add_south_China_sea(pos=[l + w - 0.091, b, .1, .2])
small_city = False
if (map_extent2[1] - map_extent2[0] < 25):
small_city = True
if city:
utl.add_city_on_map(ax,
map_extent=map_extent2,
transform=datacrs,
zorder=110,
size=13,
small_city=small_city)
utl.add_logo_extra_in_axes(pos=[l - 0.02, b + h - 0.1, .1, .1],
which='nmc',
size='Xlarge')
# show figure
if (output_dir != None):
plt.savefig(output_dir + 'Miller_综合图_预报_' + '起报时间_' +
initTime.strftime("%Y年%m月%d日%H时") + '预报时效_' +
str(fcst_info['forecast_period']) + '小时' + '.png',
dpi=200,
bbox_inches='tight')
plt.close()
if (output_dir == None):
plt.show()
def Horizontal_Pallete(figsize=(16, 9),
plotcrs=ccrs.PlateCarree(),
datacrs=ccrs.PlateCarree(),
map_extent=(60, 145, 15, 55),
title='',
forcast_info='',
add_china=True,
add_city=False,
add_background=True,
south_China_sea=True,
info_zorder=10,
title_fontsize=30):
plt.rcParams['font.sans-serif'] = ['SimHei'] # 步骤一（替换sans-serif字体）
plt.rcParams['axes.unicode_minus'] = False # 步骤二（解决坐标轴负数的负号显示问题）
if (sys.platform[0:3] == 'lin'):
locale.setlocale(locale.LC_CTYPE, 'zh_CN.utf8')
if (sys.platform[0:3] == 'win'):
locale.setlocale(locale.LC_CTYPE, 'chinese')
fig = plt.figure(figsize=figsize)
if (plotcrs is None):
plotcrs = ccrs.AlbersEqualArea(
central_latitude=(map_extent[2] + map_extent[3]) / 2.,
central_longitude=(map_extent[0] + map_extent[1]) / 2.,
standard_parallels=[30., 60.])
ax = plt.axes([0.01, 0.1, .98, .84], projection=plotcrs)
#ax.set_extent(map_extent, crs=crs)
map_extent = adjust_map_ratio(ax, map_extent=map_extent, datacrs=datacrs)
plt.title(title, loc='left', fontsize=title_fontsize)
# add grid lines
gl = ax.gridlines(crs=datacrs,
linewidth=2,
color='gray',
alpha=0.5,
linestyle='--',
zorder=info_zorder)
gl.xlocator = mpl.ticker.FixedLocator(np.arange(0, 360, 15))
gl.ylocator = mpl.ticker.FixedLocator(np.arange(-90, 90, 15))
if add_china:
utl.add_china_map_2cartopy_public(ax,
name='coastline',
edgecolor='gray',
lw=0.8,
zorder=info_zorder,
alpha=0.5)
utl.add_china_map_2cartopy_public(ax,
name='province',
edgecolor='gray',
lw=0.5,
zorder=info_zorder)
utl.add_china_map_2cartopy_public(ax,
name='nation',
edgecolor='black',
lw=0.8,
zorder=info_zorder)
utl.add_china_map_2cartopy_public(ax,
name='river',
edgecolor='#74b9ff',
lw=0.8,
zorder=info_zorder,
alpha=0.5)
if add_city:
small_city = False
if (map_extent[1] - map_extent[0] < 25):
small_city = True
utl.add_city_on_map(ax,
map_extent=map_extent,
size=12,
transform=datacrs,
zorder=info_zorder + 1,
small_city=small_city)
if add_background:
ax.add_feature(cfeature.OCEAN)
utl.add_cartopy_background(ax, name='RD')
if south_China_sea:
l, b, w, h = ax.get_position().bounds
utl.add_south_China_sea(pos=[l + w - 0.0875, b, .1, .2])
if forcast_info:
l, b, w, h = ax.get_position().bounds
bax = plt.axes([l, b + h - 0.1, .25, .1], facecolor='#FFFFFFCC')
bax.set_yticks([])
bax.set_xticks([])
bax.axis([0, 10, 0, 10])
bax.text(
2.2,
9.8,
forcast_info,
size=15,
va='top',
ha='left',
)
l, b, w, h = ax.get_position().bounds
utl.add_logo_extra_in_axes(pos=[l - 0.022, b + h - 0.1, .1, .1],
which='nmc',
size='Xlarge')
return fig, ax, map_extent
def draw_T_2m(T_2m=None,
map_extent=(50, 150, 0, 65),
regrid_shape=20,
add_china=True,
city=True,
south_China_sea=True,
output_dir=None,
Global=False):
# set font
plt.rcParams['font.sans-serif'] = ['SimHei'] # 步骤一（替换sans-serif字体）
plt.rcParams['axes.unicode_minus'] = False # 步骤二（解决坐标轴负数的负号显示问题）
# set figure
plt.figure(figsize=(16, 9))
if (Global == True):
plotcrs = ccrs.Robinson(central_longitude=115.)
else:
plotcrs = ccrs.AlbersEqualArea(
central_latitude=(map_extent[2] + map_extent[3]) / 2.,
central_longitude=(map_extent[0] + map_extent[1]) / 2.,
standard_parallels=[30., 60.])
ax = plt.axes([0.01, 0.1, .98, .84], projection=plotcrs)
datacrs = ccrs.PlateCarree()
map_extent2 = utl.adjust_map_ratio(ax,
map_extent=map_extent,
datacrs=datacrs)
# define return plots
plots = {}
if T_2m is not None:
x, y = np.meshgrid(T_2m['lon'], T_2m['lat'])
z = np.squeeze(T_2m['data'])
cmap = dk_ctables.cm_temp()
cmap.set_under(color=[0, 0, 0, 0], alpha=0.0)
plots['T_2m'] = ax.pcolormesh(x,
y,
z,
cmap=cmap,
zorder=1,
transform=datacrs,
alpha=0.5,
vmin=-45,
vmax=45)
z = gaussian_filter(z, 5)
plots['T_2m_zero'] = ax.contour(x,
y,
z,
levels=[0],
colors='#232B99',
linewidths=2,
transform=datacrs,
zorder=2)
cl_zero = plt.clabel(plots['T_2m_zero'],
inline=1,
fontsize=15,
fmt='%i',
colors='#232B99')
for t in cl_zero:
t.set_path_effects([
path_effects.Stroke(linewidth=3, foreground='white'),
path_effects.Normal()
])
plots['T_2m_35'] = ax.contour(x,
y,
z,
levels=[35, 37, 40],
colors=['#FF8F00', '#FF6200', '#FF0000'],
linewidths=2,
transform=datacrs,
zorder=2)
cl_35 = plt.clabel(plots['T_2m_35'],
inline=1,
fontsize=15,
fmt='%i',
colors='#FF0000')
for t in cl_35:
t.set_path_effects([
path_effects.Stroke(linewidth=3, foreground='white'),
path_effects.Normal()
])
plt.title('[' + T_2m.attrs['model'] + ']' + ' ' + T_2m.attrs['title'],
loc='left',
fontsize=30)
ax.add_feature(cfeature.OCEAN)
utl.add_china_map_2cartopy_public(ax,
name='coastline',
edgecolor='gray',
lw=0.8,
zorder=4,
alpha=0.5)
if add_china:
utl.add_china_map_2cartopy_public(ax,
name='province',
edgecolor='gray',
lw=0.5,
zorder=4)
utl.add_china_map_2cartopy_public(ax,
name='nation',
edgecolor='black',
lw=0.8,
zorder=4)
utl.add_china_map_2cartopy_public(ax,
name='river',
edgecolor='#74b9ff',
lw=0.8,
zorder=4,
alpha=0.5)
# grid lines
gl = ax.gridlines(crs=datacrs,
linewidth=2,
color='gray',
alpha=0.5,
linestyle='--',
zorder=5)
gl.xlocator = mpl.ticker.FixedLocator(np.arange(0, 360, 15))
gl.ylocator = mpl.ticker.FixedLocator(np.arange(-90, 90, 15))
utl.add_cartopy_background(ax, name='RD')
l, b, w, h = ax.get_position().bounds
#forecast information
bax = plt.axes([l, b + h - 0.1, .25, .1], facecolor='#FFFFFFCC')
bax.set_yticks([])
bax.set_xticks([])
bax.axis([0, 10, 0, 10])
initTime = pd.to_datetime(
str(T_2m.coords['forecast_reference_time'].values)).replace(
tzinfo=None).to_pydatetime()
fcst_time = initTime + timedelta(
hours=T_2m.coords['forecast_period'].values[0])
#发布时间
if (sys.platform[0:3] == 'lin'):
locale.setlocale(locale.LC_CTYPE, 'zh_CN.utf8')
if (sys.platform[0:3] == 'win'):
locale.setlocale(locale.LC_CTYPE, 'chinese')
plt.text(2.5, 7.5, '起报时间: ' + initTime.strftime("%Y年%m月%d日%H时"), size=15)
plt.text(2.5, 5, '预报时间: ' + fcst_time.strftime("%Y年%m月%d日%H时"), size=15)
plt.text(2.5,
2.5,
'预报时效: ' + str(int(T_2m.coords['forecast_period'].values[0])) +
'小时',
size=15)
plt.text(2.5, 0.5, 'www.nmc.cn', size=15)
# add color bar
if (T_2m != None):
cax = plt.axes([l, b - 0.04, w, .02])
cb = plt.colorbar(plots['T_2m'], cax=cax, orientation='horizontal')
cb.ax.tick_params(labelsize='x-large')
cb.set_label(u'°C', size=20)
# add south China sea
if south_China_sea:
utl.add_south_China_sea(pos=[l + w - 0.091, b, .1, .2])
small_city = False
#if(map_extent2[1]-map_extent2[0] < 25):
# small_city=True
#if city:
utl.add_city_and_number_on_map(ax,
data=T_2m,
map_extent=map_extent2,
transform=datacrs,
zorder=6,
size=13,
small_city=small_city)
utl.add_logo_extra_in_axes(pos=[l - 0.02, b + h - 0.1, .1, .1],
which='nmc',
size='Xlarge')
# show figure
if (output_dir != None):
plt.savefig(output_dir + '最低温度_预报_' + '起报时间_' +
initTime.strftime("%Y年%m月%d日%H时") + '预报时效_' +
str(T_2m.coords['forecast_period'].values[0]) + '小时' +
'.png',
dpi=200,
bbox_inches='tight')
plt.close()
if (output_dir == None):
plt.show()
def draw_low_level_wind(uv=None,
wsp=None,
map_extent=(50, 150, 0, 65),
regrid_shape=20,
add_china=True,
city=True,
south_China_sea=True,
output_dir=None,
Global=False):
plt.rcParams['font.sans-serif'] = ['SimHei'] # 步骤一（替换sans-serif字体）
plt.rcParams['axes.unicode_minus'] = False # 步骤二（解决坐标轴负数的负号显示问题）
# draw figure
plt.figure(figsize=(16, 9))
# set data projection
if (Global == True):
plotcrs = ccrs.Robinson(central_longitude=115.)
else:
plotcrs = ccrs.AlbersEqualArea(
central_latitude=(map_extent[2] + map_extent[3]) / 2.,
central_longitude=(map_extent[0] + map_extent[1]) / 2.,
standard_parallels=[30., 60.])
ax = plt.axes([0.01, 0.1, .98, .84], projection=plotcrs)
plt.title('[' + uv['model'] + '] ' + uv['lev'] + ' 风场, 风速',
loc='left',
fontsize=30)
datacrs = ccrs.PlateCarree()
#adapt to the map ratio
map_extent2 = utl.adjust_map_ratio(ax,
map_extent=map_extent,
datacrs=datacrs)
#adapt to the map ratio
ax.add_feature(cfeature.OCEAN)
utl.add_china_map_2cartopy_public(ax,
name='coastline',
edgecolor='gray',
lw=0.8,
zorder=105,
alpha=0.5)
if add_china:
utl.add_china_map_2cartopy_public(ax,
name='province',
edgecolor='gray',
lw=0.5,
zorder=105)
utl.add_china_map_2cartopy_public(ax,
name='nation',
edgecolor='black',
lw=0.8,
zorder=105)
utl.add_china_map_2cartopy_public(ax,
name='river',
edgecolor='#74b9ff',
lw=0.8,
zorder=105,
alpha=0.5)
# define return plots
plots = {}
# draw mean sea level pressure
if wsp is not None:
x, y = np.meshgrid(wsp['lon'], wsp['lat'])
z = np.squeeze(wsp['data'])
cmap = dk_ctables.cm_wind_speed_nws()
cmap.set_under(color=[0, 0, 0, 0], alpha=0.0)
plots['wsp'] = ax.pcolormesh(x,
y,
z,
cmap=cmap,
zorder=90,
transform=datacrs,
alpha=0.5)
# draw -hPa wind bards
if uv is not None:
x, y = np.meshgrid(uv['lon'], uv['lat'])
u = np.squeeze(uv['udata']) * 2.5
v = np.squeeze(uv['vdata']) * 2.5
#plots['uv'] = ax.barbs(
# x, y, u, v, length=6, regrid_shape=regrid_shape,
# transform=datacrs, fill_empty=False, sizes=dict(emptybarb=0.05),
# zorder=100)
x, y = np.meshgrid(uv['lon'], uv['lat'])
lw = 5 * wsp['data'] / wsp['data'].max()
plots['stream'] = ax.streamplot(x,
y,
uv['udata'],
uv['vdata'],
density=2,
color='r',
linewidth=lw,
zorder=100,
transform=datacrs)
spd_rtio = 0.03
for i in range(0, len(uv['lon']) - 1, 10):
for j in range(0, len(uv['lat']) - 1, 10):
ax.arrow(x[j, i],
y[j, i],
u[j, i] * spd_rtio,
v[j, i] * spd_rtio,
color='black',
zorder=100,
transform=datacrs,
width=0.05)
#ax.quiver(x[j,i], y[j,i], x[j,i]+u[j,i]*spd_rtio, y[j,i]+v[j,i]*spd_rtio, color='black', zorder=100, transform=datacrs)
# grid lines
gl = ax.gridlines(crs=datacrs,
linewidth=2,
color='gray',
alpha=0.5,
linestyle='--',
zorder=40)
gl.xlocator = mpl.ticker.FixedLocator(np.arange(0, 360, 15))
gl.ylocator = mpl.ticker.FixedLocator(np.arange(-90, 90, 15))
#http://earthpy.org/cartopy_backgroung.html
#C:\ProgramData\Anaconda3\Lib\site-packages\cartopy\data\raster\natural_earth
ax.background_img(name='RD', resolution='high')
#forecast information
bax = plt.axes([0.01, 0.835, .25, .1], facecolor='#FFFFFFCC')
bax.set_yticks([])
bax.set_xticks([])
bax.axis([0, 10, 0, 10])
initial_time = pd.to_datetime(str(
uv['init_time'])).replace(tzinfo=None).to_pydatetime()
fcst_time = initial_time + timedelta(hours=uv['fhour'])
#发布时间
if (sys.platform[0:3] == 'lin'):
locale.setlocale(locale.LC_CTYPE, 'zh_CN')
if (sys.platform[0:3] == 'win'):
locale.setlocale(locale.LC_CTYPE, 'chinese')
plt.text(2.5,
7.5,
'起报时间: ' + initial_time.strftime("%Y年%m月%d日%H时"),
size=15)
plt.text(2.5, 5, '预报时间: ' + fcst_time.strftime("%Y年%m月%d日%H时"), size=15)
plt.text(2.5, 2.5, '预报时效: ' + str(uv['fhour']) + '小时', size=15)
plt.text(2.5, 0.5, 'www.nmc.cn', size=15)
# add color bar
if (wsp != None):
cax = plt.axes([0.11, 0.06, .86, .02])
cb = plt.colorbar(plots['wsp'],
cax=cax,
orientation='horizontal',
extend='max',
extendrect=False)
cb.ax.tick_params(labelsize='x-large')
cb.set_label('(m/s)', size=20)
# add south China sea
if south_China_sea:
utl.add_south_China_sea(pos=[0.85, 0.13, .1, .2])
small_city = False
if (map_extent2[1] - map_extent2[0] < 25):
small_city = True
if city:
utl.add_city_on_map(ax,
map_extent=map_extent2,
transform=datacrs,
zorder=110,
size=13,
small_city=small_city)
utl.add_logo_extra_in_axes(pos=[-0.01, 0.835, .1, .1],
which='nmc',
size='Xlarge')
# show figure
if (output_dir != None):
plt.savefig(output_dir + '低层风_预报_' + '起报时间_' +
initial_time.strftime("%Y年%m月%d日%H时") + '预报时效_' +
str(uv['fhour']) + '小时' + '.png',
dpi=200)
if (output_dir == None):
plt.show()
def draw_PV_Div_uv(pv=None,
uv=None,
div=None,
map_extent=(50, 150, 0, 65),
regrid_shape=20,
add_china=True,
city=False,
south_China_sea=True,
output_dir=None,
Global=False):
plt.rcParams['font.sans-serif'] = ['SimHei'] # 步骤一（替换sans-serif字体）
plt.rcParams['axes.unicode_minus'] = False # 步骤二（解决坐标轴负数的负号显示问题）
# draw figure
plt.figure(figsize=(16, 9))
# set data projection
if (Global == True):
plotcrs = ccrs.Robinson(central_longitude=115.)
else:
plotcrs = ccrs.AlbersEqualArea(
central_latitude=(map_extent[2] + map_extent[3]) / 2.,
central_longitude=(map_extent[0] + map_extent[1]) / 2.,
standard_parallels=[30., 60.])
ax = plt.axes([0.01, 0.1, .98, .84], projection=plotcrs)
plt.title('[' + pv.attrs['model'] + '] ' + str(int(pv['level'])) +
'hPa 位涡扰动, 风场, 散度',
loc='left',
fontsize=30)
datacrs = ccrs.PlateCarree()
#adapt to the map ratio
map_extent2 = utl.adjust_map_ratio(ax,
map_extent=map_extent,
datacrs=datacrs)
#adapt to the map ratio
ax.add_feature(cfeature.OCEAN)
utl.add_china_map_2cartopy_public(ax,
name='coastline',
edgecolor='gray',
lw=0.8,
zorder=5,
alpha=0.5)
if add_china:
utl.add_china_map_2cartopy_public(ax,
name='province',
edgecolor='gray',
lw=0.5,
zorder=5)
utl.add_china_map_2cartopy_public(ax,
name='nation',
edgecolor='black',
lw=0.8,
zorder=5)
utl.add_china_map_2cartopy_public(ax,
name='river',
edgecolor='#74b9ff',
lw=0.8,
zorder=5,
alpha=0.5)
# define return plots
plots = {}
# draw mean sea level pressure
if div is not None:
x, y = np.meshgrid(div['lon'], div['lat'])
z = np.squeeze(div['data'])
clevs_div = np.arange(-15, 16, 1)
# plots['div'] = ax.contourf(
# x, y, z*1e5,clevs_div,cmap=plt.cm.PuOr,
# transform=datacrs,alpha=0.5, zorder=1,extend='both')
cmap = dk_ctables2.ncl_cmaps('hotcolr_19lev')
plots['div'] = ax.contourf(x,
y,
z * 1e5,
levels=clevs_div[:],
vmin=-15,
vmax=15,
cmap=cmap,
transform=datacrs,
alpha=0.5,
zorder=1,
extend='both')
# draw -hPa wind bards
if uv is not None:
x, y = np.meshgrid(uv['lon'], uv['lat'])
u = np.squeeze(uv['u'].values) * 2.5
v = np.squeeze(uv['v'].values) * 2.5
plots['uv'] = ax.barbs(x,
y,
u,
v,
length=6,
regrid_shape=regrid_shape,
transform=datacrs,
fill_empty=False,
sizes=dict(emptybarb=0.05),
zorder=2)
# draw -hPa geopotential height
if pv is not None:
x, y = np.meshgrid(pv['lon'], pv['lat'])
clevs_pv = np.arange(4, 25, 1)
plots['pv'] = ax.contour(x,
y,
np.squeeze(pv['data']) * 1e6,
clevs_pv,
colors='black',
linewidths=2,
transform=datacrs,
zorder=3)
plt.clabel(plots['pv'],
inline=1,
fontsize=20,
fmt='%.0f',
colors='black')
# grid lines
gl = ax.gridlines(crs=datacrs,
linewidth=2,
color='gray',
alpha=0.5,
linestyle='--',
zorder=4)
gl.xlocator = mpl.ticker.FixedLocator(np.arange(0, 360, 15))
gl.ylocator = mpl.ticker.FixedLocator(np.arange(-90, 90, 15))
utl.add_cartopy_background(ax, name='RD')
l, b, w, h = ax.get_position().bounds
#forecast information
bax = plt.axes([l, b + h - 0.1, .25, .1], facecolor='#FFFFFFCC')
bax.set_yticks([])
bax.set_xticks([])
bax.axis([0, 10, 0, 10])
initTime = pd.to_datetime(str(
pv['forecast_reference_time'].values)).replace(
tzinfo=None).to_pydatetime()
fcst_time = initTime + timedelta(hours=pv['forecast_period'].values[0])
#发布时间
if (sys.platform[0:3] == 'lin'):
locale.setlocale(locale.LC_CTYPE, 'zh_CN.utf8')
if (sys.platform[0:3] == 'win'):
locale.setlocale(locale.LC_CTYPE, 'chinese')
plt.text(2.5, 7.5, '起报时间: ' + initTime.strftime("%Y年%m月%d日%H时"), size=15)
plt.text(2.5, 5, '预报时间: ' + fcst_time.strftime("%Y年%m月%d日%H时"), size=15)
plt.text(2.5,
2.5,
'预报时效: ' + str(int(pv.coords['forecast_period'].values[0])) +
'小时',
size=15)
plt.text(2.5, 0.5, 'www.nmc.cn', size=15)
# add color bar
if (div != None):
cax = plt.axes([l, b - 0.04, w, .02])
cb = plt.colorbar(plots['div'],
cax=cax,
orientation='horizontal',
ticks=clevs_div[:],
extend='both',
extendrect=False)
cb.ax.tick_params(labelsize='x-large')
cb.set_label('Divergence ($10^5$ s$^{-1}$)', size=20)
# add south China sea
if south_China_sea:
utl.add_south_China_sea(pos=[l + w - 0.091, b, .1, .2])
small_city = False
if (map_extent2[1] - map_extent2[0] < 25):
small_city = True
if city:
utl.add_city_on_map(ax,
map_extent=map_extent2,
transform=datacrs,
zorder=110,
size=13,
small_city=small_city)
utl.add_logo_extra_in_axes(pos=[l - 0.02, b + h - 0.1, .1, .1],
which='nmc',
size='Xlarge')
# show figure
if (output_dir != None):
plt.savefig(output_dir + '位涡_风场_散度_预报_' + '起报时间_' +
initTime.strftime("%Y年%m月%d日%H时") + '预报时效_' +
str(pv.coords['forecast_period'].values[0]) + '小时' +
'.png',
dpi=200,
bbox_inches='tight')
plt.close()
if (output_dir == None):
plt.show()
def draw_mslp_rain_snow(
rain=None, snow=None,sleet=None,mslp=None,
map_extent=(50, 150, 0, 65),
regrid_shape=20,
add_china=True,city=True,south_China_sea=True,
output_dir=None,Global=False):
# set font
plt.rcParams['font.sans-serif'] = ['SimHei'] # 步骤一（替换sans-serif字体）
plt.rcParams['axes.unicode_minus'] = False # 步骤二（解决坐标轴负数的负号显示问题）
# set figure
plt.figure(figsize=(16,9))
if(Global == True):
plotcrs = ccrs.Robinson(central_longitude=115.)
else:
plotcrs = ccrs.AlbersEqualArea(central_latitude=(map_extent[2]+map_extent[3])/2.,
central_longitude=(map_extent[0]+map_extent[1])/2., standard_parallels=[30., 60.])
ax = plt.axes([0.01,0.1,.98,.84], projection=plotcrs)
datacrs = ccrs.PlateCarree()
map_extent2=utl.adjust_map_ratio(ax,map_extent=map_extent,datacrs=datacrs)
plt.title('['+mslp.attrs['model']+'] '+
'海平面气压, '+
str(rain.attrs['atime'])+'小时降水',
loc='left', fontsize=30)
#draw data
plots = {}
if rain is not None:
x, y = np.meshgrid(rain['lon'], rain['lat'])
z=np.squeeze(rain.values)
cmap,norm=dk_ctables.cm_rain_nws(atime=rain.attrs['atime'])
#cmap.set_under(color=[0,0,0,0],alpha=0.0)
plots['rain'] = ax.pcolormesh(
x,y,z, norm=norm,
cmap=cmap, zorder=3,transform=datacrs,alpha=0.8)
if snow is not None:
x, y = np.meshgrid(snow['lon'], snow['lat'])
z=np.squeeze(snow.values)
cmap,norm=dk_ctables.cm_snow_nws(atime=rain.attrs['atime'])
#cmap.set_under(color=[0,0,0,0],alpha=0.0)
plots['snow'] = ax.pcolormesh(
x,y,z, norm=norm,
cmap=cmap, zorder=3,transform=datacrs,alpha=0.8)
if sleet is not None:
x, y = np.meshgrid(sleet['lon'], sleet['lat'])
z=np.squeeze(sleet.values)
cmap,norm=dk_ctables.cm_sleet_nws(atime=rain.attrs['atime'])
#cmap.set_under(color=[0,0,0,0],alpha=0.0)
plots['sleet'] = ax.pcolormesh(
x,y,z, norm=norm,
cmap=cmap, zorder=3,transform=datacrs,alpha=0.8)
if mslp is not None:
x, y = np.meshgrid(mslp['lon'], mslp['lat'])
clevs_mslp = np.arange(900, 1100,2.5)
z=gaussian_filter(np.squeeze(mslp['data']), 5)
plots['mslp'] = ax.contour(
x, y, z, clevs_mslp, colors='black',
linewidths=2, transform=datacrs, zorder=3)
plt.clabel(plots['mslp'], inline=1, fontsize=20, fmt='%.0f',colors='black')
#additional information
ax.add_feature(cfeature.OCEAN)
utl.add_china_map_2cartopy_public(
ax, name='coastline', edgecolor='gray', lw=0.8, zorder=1,alpha=0.5)
if add_china:
utl.add_china_map_2cartopy_public(
ax, name='province', edgecolor='gray', lw=0.5, zorder=1)
utl.add_china_map_2cartopy_public(
ax, name='nation', edgecolor='black', lw=0.8, zorder=1)
utl.add_china_map_2cartopy_public(
ax, name='river', edgecolor='#74b9ff', lw=0.8, zorder=1,alpha=0.5)
# grid lines
gl = ax.gridlines(
crs=datacrs, linewidth=2, color='gray', alpha=0.5, linestyle='--', zorder=1)
gl.xlocator = mpl.ticker.FixedLocator(np.arange(0, 360, 15))
gl.ylocator = mpl.ticker.FixedLocator(np.arange(-90, 90, 15))
# utl.add_cartopy_background(ax,name='RD')
ax.add_feature(cfeature.LAND,color='#F5E19F')
ax.add_feature(cfeature.OCEAN)
l, b, w, h = ax.get_position().bounds
#forecast information
bax=plt.axes([l,b+h-0.1,.25,.1],facecolor='#FFFFFFCC')
bax.set_yticks([])
bax.set_xticks([])
bax.axis([0, 10, 0, 10])
initTime = pd.to_datetime(
str(mslp.coords['forecast_reference_time'].values)).replace(tzinfo=None).to_pydatetime()
fcst_time=initTime+timedelta(hours=mslp.coords['forecast_period'].values[0])
#发布时间
if(sys.platform[0:3] == 'lin'):
locale.setlocale(locale.LC_CTYPE, 'zh_CN.utf8')
if(sys.platform[0:3] == 'win'):
locale.setlocale(locale.LC_CTYPE, 'chinese')
plt.text(2.5, 7.5,'起报时间: '+initTime.strftime("%Y年%m月%d日%H时"),size=15)
plt.text(2.5, 5,'预报时间: '+fcst_time.strftime("%Y年%m月%d日%H时"),size=15)
plt.text(2.5, 2.5,'预报时效: '+str(int(mslp.coords['forecast_period'].values[0]))+'小时',size=15)
plt.text(2.5, 0.5,'www.nmc.cn',size=15)
# add color bar
if(sleet is not None):
cax=plt.axes([l-0.03,b-0.04,w/3,.02])
cb = plt.colorbar(plots['sleet'], cax=cax, orientation='horizontal')
cb.ax.tick_params(labelsize='x-large')
cb.set_label('雨夹雪 (mm)',size=20)
if(snow is not None):
# cax=plt.axes([l+0.32,b-0.04,w/4,.02])
cax=plt.axes([l+w/3,b-0.04,w/3,.02])
cb = plt.colorbar(plots['snow'], cax=cax, orientation='horizontal')
cb.ax.tick_params(labelsize='x-large')
cb.set_label('雪 (mm)',size=20)
if(rain is not None):
# cax=plt.axes([l+0.65,b-0.04,w/4,.02])
cax=plt.axes([l+w*2/3+0.03,b-0.04,w/3,.02])
cb = plt.colorbar(plots['rain'], cax=cax, orientation='horizontal')
cb.ax.tick_params(labelsize='x-large')
cb.set_label('雨 (mm)',size=20)
# add south China sea
if south_China_sea:
utl.add_south_China_sea(pos=[l+w-0.091,b,.1,.2])
small_city=False
if(map_extent2[1]-map_extent2[0] < 25):
small_city=True
if city:
utl.add_city_on_map(ax,map_extent=map_extent2,transform=datacrs,zorder=110,size=13,small_city=small_city)
utl.add_logo_extra_in_axes(pos=[l-0.02,b+h-0.1,.1,.1],which='nmc', size='Xlarge')
# show figure
if(output_dir != None):
plt.savefig(output_dir+'海平面气压_降水_预报_'+
'起报时间_'+initTime.strftime("%Y年%m月%d日%H时")+
'预报时效_'+str(int(mslp.coords['forecast_period'].values[0]))+'小时'+'.png',
dpi=200,bbox_inches='tight')
plt.close()
if(output_dir == None):
plt.show()
def draw_T_2m(T_2m=None,
map_extent=(50, 150, 0, 65),
regrid_shape=20,
add_china=True,
city=True,
south_China_sea=True,
output_dir=None,
Global=False):
plt.rcParams['font.sans-serif'] = ['SimHei'] # 步骤一（替换sans-serif字体）
plt.rcParams['axes.unicode_minus'] = False # 步骤二（解决坐标轴负数的负号显示问题）
# draw figure
plt.figure(figsize=(16, 9))
# set data projection
if (Global == True):
plotcrs = ccrs.Robinson(central_longitude=115.)
else:
plotcrs = ccrs.AlbersEqualArea(
central_latitude=(map_extent[2] + map_extent[3]) / 2.,
central_longitude=(map_extent[0] + map_extent[1]) / 2.,
standard_parallels=[30., 60.])
ax = plt.axes([0.01, 0.1, .98, .84], projection=plotcrs)
plt.title('[' + T_2m['model'] + ']' + ' ' + T_2m['title'],
loc='left',
fontsize=30)
datacrs = ccrs.PlateCarree()
#adapt to the map ratio
map_extent2 = utl.adjust_map_ratio(ax,
map_extent=map_extent,
datacrs=datacrs)
#adapt to the map ratio
ax.add_feature(cfeature.OCEAN)
utl.add_china_map_2cartopy_public(ax,
name='coastline',
edgecolor='gray',
lw=0.8,
zorder=105,
alpha=0.5)
if add_china:
utl.add_china_map_2cartopy_public(ax,
name='province',
edgecolor='gray',
lw=0.5,
zorder=105)
utl.add_china_map_2cartopy_public(ax,
name='nation',
edgecolor='black',
lw=0.8,
zorder=105)
utl.add_china_map_2cartopy_public(ax,
name='river',
edgecolor='#74b9ff',
lw=0.8,
zorder=105,
alpha=0.5)
# define return plots
plots = {}
# draw mean sea level pressure
if T_2m is not None:
x, y = np.meshgrid(T_2m['lon'], T_2m['lat'])
z = np.squeeze(T_2m['data'])
cmap = dk_ctables.cm_temperature_nws()
cmap.set_under(color=[0, 0, 0, 0], alpha=0.0)
plots['T_2m'] = ax.pcolormesh(x,
y,
z,
cmap=cmap,
zorder=100,
transform=datacrs,
alpha=0.5,
vmin=-45,
vmax=45)
clevs_zero = 0
z = gaussian_filter(z, 5)
plots['T_2m_zero'] = ax.contour(x,
y,
z,
clevs_zero,
colors='red',
linewidths=2,
transform=datacrs,
zorder=110)
# grid lines
gl = ax.gridlines(crs=datacrs,
linewidth=2,
color='gray',
alpha=0.5,
linestyle='--',
zorder=120)
gl.xlocator = mpl.ticker.FixedLocator(np.arange(0, 360, 15))
gl.ylocator = mpl.ticker.FixedLocator(np.arange(-90, 90, 15))
#http://earthpy.org/cartopy_backgroung.html
#C:\ProgramData\Anaconda3\Lib\site-packages\cartopy\data\raster\natural_earth
ax.background_img(name='RD', resolution='high')
#forecast information
bax = plt.axes([0.01, 0.835, .25, .1], facecolor='#FFFFFFCC')
bax.set_yticks([])
bax.set_xticks([])
bax.axis([0, 10, 0, 10])
initial_time = pd.to_datetime(str(
T_2m['init_time'])).replace(tzinfo=None).to_pydatetime()
fcst_time = initial_time + timedelta(hours=T_2m['fhour'])
#发布时间
if (sys.platform[0:3] == 'lin'):
locale.setlocale(locale.LC_CTYPE, 'zh_CN')
if (sys.platform[0:3] == 'win'):
locale.setlocale(locale.LC_CTYPE, 'chinese')
plt.text(2.5,
7.5,
'起报时间: ' + initial_time.strftime("%Y年%m月%d日%H时"),
size=15)
plt.text(2.5, 5, '预报时间: ' + fcst_time.strftime("%Y年%m月%d日%H时"), size=15)
plt.text(2.5, 2.5, '预报时效: ' + str(T_2m['fhour']) + '小时', size=15)
plt.text(2.5, 0.5, 'www.nmc.cn', size=15)
# add color bar
if (T_2m != None):
cax = plt.axes([0.11, 0.06, .86, .02])
cb = plt.colorbar(plots['T_2m'], cax=cax, orientation='horizontal')
cb.ax.tick_params(labelsize='x-large')
cb.set_label(u'°C', size=20)
# add south China sea
if south_China_sea:
utl.add_south_China_sea(pos=[0.85, 0.13, .1, .2])
small_city = False
if (map_extent2[1] - map_extent2[0] < 25):
small_city = True
if city:
utl.add_city_on_map(ax,
map_extent=map_extent2,
transform=datacrs,
zorder=110,
size=13,
small_city=small_city)
utl.add_logo_extra_in_axes(pos=[-0.01, 0.835, .1, .1],
which='nmc',
size='Xlarge')
# show figure
if (output_dir != None):
plt.savefig(output_dir + '最低温度_预报_' + '起报时间_' +
initial_time.strftime("%Y年%m月%d日%H时") + '预报时效_' +
str(T_2m['fhour']) + '小时' + '.png',
dpi=200)
if (output_dir == None):
plt.show()
def draw_gh_uv_r6(gh=None, uv=None, r6=None,
map_extent=(50, 150, 0, 65),
regrid_shape=20,
add_china=True,city=True,south_China_sea=True,
output_dir=None,Global=False):
"""
Draw -hPa geopotential height contours, -hPa wind barbs
and mean sea level pressure filled contours.
:param gh: -hPa gh, dictionary:
necessary, {'lon': 1D array, 'lat': 1D array,
'data': 2D array}
optional, {'clevs': 1D array}
:param uv: -hPa u-component and v-component wind, dictionary:
necessary, {'lon': 1D array, 'lat': 1D array,
'udata': 2D array, 'vdata': 2D array}
:param r6: r6, dictionary:
necessary, {'lon': 1D array, 'lat': 1D array,
'data': 2D array}
optional, {'clevs': 1D array}
:param map_extent: [lonmin, lonmax, latmin, latmax],
longitude and latitude range.
:param add_china: add china map or not.
:param regrid_shape: control the wind barbs density.
:return: plots dictionary.
:Examples:
"""
plt.rcParams['font.sans-serif'] = ['SimHei'] # 步骤一（替换sans-serif字体）
plt.rcParams['axes.unicode_minus'] = False # 步骤二（解决坐标轴负数的负号显示问题）
# draw figure
plt.figure(figsize=(16,9))
# set data projection
if(Global == True):
plotcrs = ccrs.Robinson(central_longitude=115.)
else:
plotcrs = ccrs.AlbersEqualArea(central_latitude=(map_extent[2]+map_extent[3])/2.,
central_longitude=(map_extent[0]+map_extent[1])/2., standard_parallels=[30., 60.])
ax = plt.axes([0.01,0.1,.98,.84], projection=plotcrs)
plt.title('['+gh['model']+'] '+
gh['lev']+'hPa 位势高度场, '+
uv['lev']+'hPa 风场, 6小时降水',
loc='left', fontsize=30)
datacrs = ccrs.PlateCarree()
#adapt to the map ratio
map_extent2=utl.adjust_map_ratio(ax,map_extent=map_extent,datacrs=datacrs)
#adapt to the map ratio
ax.add_feature(cfeature.OCEAN)
utl.add_china_map_2cartopy_public(
ax, name='coastline', edgecolor='gray', lw=0.8, zorder=105,alpha=0.5)
if add_china:
utl.add_china_map_2cartopy_public(
ax, name='province', edgecolor='gray', lw=0.5, zorder=105)
utl.add_china_map_2cartopy_public(
ax, name='nation', edgecolor='black', lw=0.8, zorder=105)
utl.add_china_map_2cartopy_public(
ax, name='river', edgecolor='#74b9ff', lw=0.8, zorder=105,alpha=0.5)
# define return plots
plots = {}
# draw mean sea level pressure
if r6 is not None:
x, y = np.meshgrid(r6['lon'], r6['lat'])
clevs_r6 = [0.1, 4, 13, 25, 60, 120]
plots['r6'] = ax.contourf(
x, y, np.squeeze(r6['data']), clevs_r6,
colors=["#88F492", "#00A929", "#2AB8FF", "#1202FC", "#FF04F4", "#850C3E"],
alpha=0.8, zorder=10, transform=datacrs,extend='max',extendrect=False)
# draw -hPa wind bards
if uv is not None:
x, y = np.meshgrid(uv['lon'], uv['lat'])
u = np.squeeze(uv['udata']) * 2.5
v = np.squeeze(uv['vdata']) * 2.5
plots['uv'] = ax.barbs(
x, y, u, v, length=6, regrid_shape=regrid_shape,
transform=datacrs, fill_empty=False, sizes=dict(emptybarb=0.05),
zorder=20)
# draw -hPa geopotential height
if gh is not None:
x, y = np.meshgrid(gh['lon'], gh['lat'])
clevs_gh = np.append(np.append(np.arange(0, 480, 4),np.append(np.arange(480, 584, 8), np.arange(580, 604, 4))), np.arange(604, 2000, 8))
plots['gh'] = ax.contour(
x, y, np.squeeze(gh['data']), clevs_gh, colors='black',
linewidths=2, transform=datacrs, zorder=30)
plt.clabel(plots['gh'], inline=1, fontsize=20, fmt='%.0f',colors='black')
# grid lines
gl = ax.gridlines(
crs=datacrs, linewidth=2, color='gray', alpha=0.5, linestyle='--', zorder=40)
gl.xlocator = mpl.ticker.FixedLocator(np.arange(0, 360, 15))
gl.ylocator = mpl.ticker.FixedLocator(np.arange(-90, 90, 15))
#http://earthpy.org/cartopy_backgroung.html
#C:\ProgramData\Anaconda3\Lib\site-packages\cartopy\data\raster\natural_earth
ax.background_img(name='RD', resolution='high')
#forecast information
bax=plt.axes([0.01,0.835,.25,.1],facecolor='#FFFFFFCC')
bax.set_yticks([])
bax.set_xticks([])
bax.axis([0, 10, 0, 10])
initial_time = pd.to_datetime(
str(gh['init_time'])).replace(tzinfo=None).to_pydatetime()
fcst_time=initial_time+timedelta(hours=gh['fhour'])
#发布时间
if(sys.platform[0:3] == 'lin'):
locale.setlocale(locale.LC_CTYPE, 'zh_CN')
if(sys.platform[0:3] == 'win'):
locale.setlocale(locale.LC_CTYPE, 'chinese')
plt.text(2.5, 7.5,'起报时间: '+initial_time.strftime("%Y年%m月%d日%H时"),size=15)
plt.text(2.5, 5,'预报时间: '+fcst_time.strftime("%Y年%m月%d日%H时"),size=15)
plt.text(2.5, 2.5,'预报时效: '+str(gh['fhour'])+'小时',size=15)
plt.text(2.5, 0.5,'www.nmc.cn',size=15)
# add color bar
if(r6 != None):
cax=plt.axes([0.11,0.06,.86,.02])
cb = plt.colorbar(plots['r6'], cax=cax, orientation='horizontal',
ticks=clevs_r6[:],
extend='max',extendrect=False)
cb.ax.tick_params(labelsize='x-large')
cb.set_label('6h precipitation (mm)',size=20)
# add south China sea
if south_China_sea:
utl.add_south_China_sea(pos=[0.85,0.13,.1,.2])
small_city=False
if(map_extent2[1]-map_extent2[0] < 25):
small_city=True
if city:
utl.add_city_on_map(ax,map_extent=map_extent2,transform=datacrs,zorder=110,size=13,small_city=small_city)
utl.add_logo_extra_in_axes(pos=[-0.01,0.835,.1,.1],which='nmc', size='Xlarge')
# show figure
if(output_dir != None):
plt.savefig(output_dir+'高度场_风场_降水_预报_'+
'起报时间_'+initial_time.strftime("%Y年%m月%d日%H时")+
'预报时效_'+str(gh['fhour'])+'小时'+'.png', dpi=200)
if(output_dir == None):
plt.show()
def draw_mslp_gust10m_uv10m(gust=None,
mslp=None,
uv10m=None,
map_extent=(50, 150, 0, 65),
regrid_shape=20,
add_china=True,
city=True,
south_China_sea=True,
output_dir=None,
Global=False):
# set font
plt.rcParams['font.sans-serif'] = ['SimHei'] # 步骤一（替换sans-serif字体）
plt.rcParams['axes.unicode_minus'] = False # 步骤二（解决坐标轴负数的负号显示问题）
# set figure
plt.figure(figsize=(16, 9))
if (Global == True):
plotcrs = ccrs.Robinson(central_longitude=115.)
else:
plotcrs = ccrs.AlbersEqualArea(
central_latitude=(map_extent[2] + map_extent[3]) / 2.,
central_longitude=(map_extent[0] + map_extent[1]) / 2.,
standard_parallels=[30., 60.])
datacrs = ccrs.PlateCarree()
ax = plt.axes([0.01, 0.1, .98, .84], projection=plotcrs)
map_extent2 = utl.adjust_map_ratio(ax,
map_extent=map_extent,
datacrs=datacrs)
#draw data
plots = {}
if gust is not None:
x, y = np.meshgrid(gust['lon'], gust['lat'])
z = np.squeeze(gust['data'].values)
cmap = dk_ctables.cm_wind_speed_nws()
cmap.set_under(color=[0, 0, 0, 0], alpha=0.0)
z[z < 7.9] = np.nan
plots['gust'] = ax.pcolormesh(x,
y,
z,
cmap=cmap,
vmin=7.9,
vmax=65,
zorder=1,
transform=datacrs,
alpha=0.5)
ax.quiver(x,
y,
np.squeeze(uv10m['u10m'].values),
np.squeeze(uv10m['v10m'].values),
transform=datacrs,
regrid_shape=40,
width=0.001)
if mslp is not None:
x, y = np.meshgrid(mslp['lon'], mslp['lat'])
clevs_mslp = np.arange(900, 1100, 2.5)
z = gaussian_filter(np.squeeze(mslp['data']), 5)
plots['mslp'] = ax.contour(x,
y,
z,
clevs_mslp,
colors='black',
linewidths=1,
transform=datacrs,
zorder=2,
alpha=0.5)
cl = plt.clabel(plots['mslp'],
inline=1,
fontsize=15,
fmt='%.1f',
colors='black')
for t in cl:
t.set_path_effects([
path_effects.Stroke(linewidth=3, foreground='white'),
path_effects.Normal()
])
#+画高低压中心
res = mslp['lon'].values[1] - mslp['lon'].values[0]
nwindow = int(9.5 / res)
mslp_hl = np.ma.masked_invalid(mslp['data'].values).squeeze()
local_min, local_max = utl.extrema(mslp_hl,
mode='wrap',
window=nwindow)
#Get location of extrema on grid
xmin, xmax, ymin, ymax = map_extent2
lons2d, lats2d = x, y
transformed = datacrs.transform_points(datacrs, lons2d, lats2d)
x = transformed[..., 0]
y = transformed[..., 1]
xlows = x[local_min]
xhighs = x[local_max]
ylows = y[local_min]
yhighs = y[local_max]
lowvals = mslp_hl[local_min]
highvals = mslp_hl[local_max]
yoffset = 0.022 * (ymax - ymin)
dmin = yoffset
#Plot low pressures
xyplotted = []
for x, y, p in zip(xlows, ylows, lowvals):
if x < xmax - yoffset and x > xmin + yoffset and y < ymax - yoffset and y > ymin + yoffset:
dist = [
np.sqrt((x - x0)**2 + (y - y0)**2) for x0, y0 in xyplotted
]
if not dist or min(dist) > dmin: #,fontweight='bold'
a = ax.text(x,
y,
'L',
fontsize=28,
ha='center',
va='center',
color='r',
fontweight='normal',
transform=datacrs)
b = ax.text(x,
y - yoffset,
repr(int(p)),
fontsize=14,
ha='center',
va='top',
color='r',
fontweight='normal',
transform=datacrs)
a.set_path_effects([
path_effects.Stroke(linewidth=1.5, foreground='black'),
path_effects.SimpleLineShadow(),
path_effects.Normal()
])
b.set_path_effects([
path_effects.Stroke(linewidth=1.0, foreground='black'),
path_effects.SimpleLineShadow(),
path_effects.Normal()
])
xyplotted.append((x, y))
#Plot high pressures
xyplotted = []
for x, y, p in zip(xhighs, yhighs, highvals):
if x < xmax - yoffset and x > xmin + yoffset and y < ymax - yoffset and y > ymin + yoffset:
dist = [
np.sqrt((x - x0)**2 + (y - y0)**2) for x0, y0 in xyplotted
]
if not dist or min(dist) > dmin:
a = ax.text(x,
y,
'H',
fontsize=28,
ha='center',
va='center',
color='b',
fontweight='normal',
transform=datacrs)
b = ax.text(x,
y - yoffset,
repr(int(p)),
fontsize=14,
ha='center',
va='top',
color='b',
fontweight='normal',
transform=datacrs)
a.set_path_effects([
path_effects.Stroke(linewidth=1.5, foreground='black'),
path_effects.SimpleLineShadow(),
path_effects.Normal()
])
b.set_path_effects([
path_effects.Stroke(linewidth=1.0, foreground='black'),
path_effects.SimpleLineShadow(),
path_effects.Normal()
])
xyplotted.append((x, y))
#-画高低压中心
#additional information
plt.title('[' + mslp.attrs['model'] + '] ' + '海平面气压, ' + '过去' +
'%i' % gust.attrs['t_gap'] + '小时10米最大阵风和10米平均风',
loc='left',
fontsize=30)
utl.add_china_map_2cartopy_public(ax,
name='coastline',
edgecolor='gray',
lw=0.8,
zorder=3,
alpha=0.5)
if add_china:
utl.add_china_map_2cartopy_public(ax,
name='province',
edgecolor='gray',
lw=0.5,
zorder=3)
utl.add_china_map_2cartopy_public(ax,
name='nation',
edgecolor='black',
lw=0.8,
zorder=3)
utl.add_china_map_2cartopy_public(ax,
name='river',
edgecolor='#74b9ff',
lw=0.8,
zorder=3,
alpha=0.5)
# grid lines
gl = ax.gridlines(crs=datacrs,
linewidth=2,
color='gray',
alpha=0.5,
linestyle='--',
zorder=4)
gl.xlocator = mpl.ticker.FixedLocator(np.arange(0, 360, 15))
gl.ylocator = mpl.ticker.FixedLocator(np.arange(-90, 90, 15))
#utl.add_cartopy_background(ax,name='RD')
ax.add_feature(cfeature.LAND, color='#F5E19F')
ax.add_feature(cfeature.OCEAN)
l, b, w, h = ax.get_position().bounds
#forecast information
bax = plt.axes([l, b + h - 0.1, .25, .1], facecolor='#FFFFFFCC')
bax.set_yticks([])
bax.set_xticks([])
bax.axis([0, 10, 0, 10])
initTime = pd.to_datetime(
str(mslp.coords['forecast_reference_time'].values)).replace(
tzinfo=None).to_pydatetime()
fcst_time = initTime + timedelta(
hours=mslp.coords['forecast_period'].values[0])
#发布时间
if (sys.platform[0:3] == 'lin'):
locale.setlocale(locale.LC_CTYPE, 'zh_CN.utf8')
if (sys.platform[0:3] == 'win'):
locale.setlocale(locale.LC_CTYPE, 'chinese')
plt.text(2.5, 7.5, '起报时间: ' + initTime.strftime("%Y年%m月%d日%H时"), size=15)
plt.text(2.5, 5, '预报时间: ' + fcst_time.strftime("%Y年%m月%d日%H时"), size=15)
plt.text(2.5,
2.5,
'预报时效: ' + str(int(mslp.coords['forecast_period'].values[0])) +
'小时',
size=15)
plt.text(2.5, 0.5, 'www.nmc.cn', size=15)
# add color bar
if (gust != None):
cax = plt.axes([l, b - 0.04, w, .02])
cb = plt.colorbar(plots['gust'],
cax=cax,
orientation='horizontal',
extend='max',
ticks=[
8.0, 10.8, 13.9, 17.2, 20.8, 24.5, 28.5, 32.7,
37, 41.5, 46.2, 51.0, 56.1, 61.3
])
cb.ax.tick_params(labelsize='x-large')
cb.set_label('风速 (m/s)', size=20)
# add south China sea
if south_China_sea:
utl.add_south_China_sea(pos=[l + w - 0.091, b, .1, .2])
small_city = False
if (map_extent2[1] - map_extent2[0] < 25):
small_city = True
if city:
utl.add_city_on_map(ax,
map_extent=map_extent2,
transform=datacrs,
zorder=5,
size=13,
small_city=small_city)
utl.add_logo_extra_in_axes(pos=[l - 0.02, b + h - 0.1, .1, .1],
which='nmc',
size='Xlarge')
# show figure
if (output_dir != None):
plt.savefig(output_dir + '海平面气压_逐6小时最大风速_预报_' + '起报时间_' +
initTime.strftime("%Y年%m月%d日%H时") + '预报时效_' +
str(mslp.coords['forecast_period'].values[0]) + '小时' +
'.png',
dpi=200,
bbox_inches='tight')
plt.close()
if (output_dir == None):
plt.show()
def draw_T2m_mslp_uv10m(t2m=None,
mslp=None,
uv10m=None,
map_extent=(50, 150, 0, 65),
regrid_shape=20,
add_china=True,
city=True,
south_China_sea=True,
output_dir=None,
Global=False):
# set font
plt.rcParams['font.sans-serif'] = ['SimHei'] # 步骤一（替换sans-serif字体）
plt.rcParams['axes.unicode_minus'] = False # 步骤二（解决坐标轴负数的负号显示问题）
# draw figure
plt.figure(figsize=(16, 9))
if (Global == True):
plotcrs = ccrs.Robinson(central_longitude=115.)
else:
plotcrs = ccrs.AlbersEqualArea(
central_latitude=(map_extent[2] + map_extent[3]) / 2.,
central_longitude=(map_extent[0] + map_extent[1]) / 2.,
standard_parallels=[30., 60.])
datacrs = ccrs.PlateCarree()
ax = plt.axes([0.01, 0.1, .98, .84], projection=plotcrs)
map_extent2 = utl.adjust_map_ratio(ax,
map_extent=map_extent,
datacrs=datacrs)
# define return plots
plots = {}
# draw mean sea level pressure
if t2m is not None:
x, y = np.meshgrid(t2m['lon'], t2m['lat'])
z = np.squeeze(t2m['data'])
cmap = dk_ctables.cm_high_temperature_nws()
cmap.set_under(color=[0, 0, 0, 0], alpha=0.0)
plots['t2m'] = ax.pcolormesh(x,
y,
z,
cmap=cmap,
zorder=1,
transform=datacrs,
alpha=0.5)
if uv10m is not None:
x, y = np.meshgrid(uv10m['lon'].values, uv10m['lat'].values)
u = np.squeeze(uv10m['u10m'].values) * 2.5
v = np.squeeze(uv10m['v10m'].values) * 2.5
plots['uv'] = ax.barbs(x,
y,
u,
v,
length=6,
regrid_shape=regrid_shape,
transform=datacrs,
fill_empty=False,
sizes=dict(emptybarb=0.05),
zorder=2,
color='white')
if mslp is not None:
x, y = np.meshgrid(mslp['lon'], mslp['lat'])
clevs_mslp = np.arange(900, 1100, 2.5)
z = gaussian_filter(np.squeeze(mslp['data']), 5)
plots['mslp'] = ax.contour(x,
y,
z,
clevs_mslp,
colors='black',
linewidths=1,
transform=datacrs,
zorder=3)
cl = ax.clabel(plots['mslp'],
inline=1,
fontsize=20,
fmt='%.0f',
colors='black')
#additional information
plt.title('[' + mslp.attrs['model'] + '] ' + '海平面气压, ' + '10米风场, ' +
'2米 温度',
loc='left',
fontsize=30)
ax.add_feature(cfeature.OCEAN)
utl.add_china_map_2cartopy_public(ax,
name='coastline',
edgecolor='gray',
lw=0.8,
zorder=4,
alpha=0.5)
if add_china:
utl.add_china_map_2cartopy_public(ax,
name='province',
edgecolor='gray',
lw=0.5,
zorder=4)
utl.add_china_map_2cartopy_public(ax,
name='nation',
edgecolor='black',
lw=0.8,
zorder=4)
utl.add_china_map_2cartopy_public(ax,
name='river',
edgecolor='#74b9ff',
lw=0.8,
zorder=4,
alpha=0.5)
# grid lines
gl = ax.gridlines(crs=datacrs,
linewidth=2,
color='gray',
alpha=0.5,
linestyle='--',
zorder=5)
gl.xlocator = mpl.ticker.FixedLocator(np.arange(0, 360, 15))
gl.ylocator = mpl.ticker.FixedLocator(np.arange(-90, 90, 15))
utl.add_cartopy_background(ax, name='RD')
l, b, w, h = ax.get_position().bounds
#forecast information
bax = plt.axes([l, b + h - 0.1, .25, .1], facecolor='#FFFFFFCC')
bax.set_yticks([])
bax.set_xticks([])
bax.axis([0, 10, 0, 10])
initTime = pd.to_datetime(str(
mslp['forecast_reference_time'].values)).replace(
tzinfo=None).to_pydatetime()
fcst_time = initTime + timedelta(
hours=mslp.coords['forecast_period'].values[0])
#发布时间
if (sys.platform[0:3] == 'lin'):
locale.setlocale(locale.LC_CTYPE, 'zh_CN.utf8')
if (sys.platform[0:3] == 'win'):
locale.setlocale(locale.LC_CTYPE, 'chinese')
plt.text(2.5, 7.5, '起报时间: ' + initTime.strftime("%Y年%m月%d日%H时"), size=15)
plt.text(2.5, 5, '预报时间: ' + fcst_time.strftime("%Y年%m月%d日%H时"), size=15)
plt.text(2.5,
2.5,
'预报时效: ' + str(mslp.coords['forecast_period'].values[0]) + '小时',
size=15)
plt.text(2.5, 0.5, 'www.nmc.cn', size=15)
# add color bar
if (t2m != None):
cax = plt.axes([l, b - 0.04, w, .02])
cb = plt.colorbar(plots['t2m'], cax=cax, orientation='horizontal')
cb.ax.tick_params(labelsize='x-large')
cb.set_label('Temperature at 2m Above Ground (' + u'°C' + ')', size=20)
# add south China sea
if south_China_sea:
utl.add_south_China_sea(pos=[l + w - 0.091, b, .1, .2])
small_city = False
if (map_extent2[1] - map_extent2[0] < 25):
small_city = True
if city:
utl.add_city_on_map(ax,
map_extent=map_extent2,
transform=datacrs,
zorder=6,
size=13,
small_city=small_city)
utl.add_logo_extra_in_axes(pos=[l - 0.02, b + h - 0.1, .1, .1],
which='nmc',
size='Xlarge')
# show figure
if (output_dir != None):
plt.savefig(output_dir + '海平面气压_10米风场_2米温度_预报_' + '起报时间_' +
initTime.strftime("%Y年%m月%d日%H时") + '预报时效_' +
str(mslp.coords['forecast_period'].values[0]) + '小时' +
'.png',
dpi=200,
bbox_inches='tight')
plt.close()
if (output_dir == None):
plt.show()
def draw_cumulated_precip(
rain=None,
map_extent=(50, 150, 0, 65),
regrid_shape=20,
add_china=True,city=True,south_China_sea=True,
output_dir=None,Global=False):
# set font
plt.rcParams['font.sans-serif'] = ['SimHei'] # 步骤一（替换sans-serif字体）
plt.rcParams['axes.unicode_minus'] = False # 步骤二（解决坐标轴负数的负号显示问题）
# set figure
fig = plt.figure(figsize=(9,14))
camera = Camera(fig)
nframe=rain['data'].shape[0]
plotcrs = ccrs.AlbersEqualArea(central_latitude=(map_extent[2]+map_extent[3])/2.,
central_longitude=(map_extent[0]+map_extent[1])/2., standard_parallels=[30., 60.])
datacrs = ccrs.PlateCarree()
ax = plt.axes([0.01,0.1,.98,.84], projection=plotcrs)
map_extent2=utl.adjust_map_ratio(ax,map_extent=map_extent,datacrs=datacrs)
# define return plots
plots = {}
# draw mean sea level pressure
if rain is not None:
x, y = np.meshgrid(rain['lon'], rain['lat'])
z=np.squeeze(rain['data'].values)
z[z<0.1]=np.nan
znan=np.zeros(shape=x.shape)
znan[:]=np.nan
# cmap,norm=dk_ctables.cm_qpf_nws(atime=None,
# pos=np.concatenate((
# np.array([0, 0.1, 0.5, 1]), np.arange(2.5, 25, 2.5),
# np.arange(25, 50, 5), np.arange(50, 150, 10),
# np.arange(150, 800, 50))))
cmap,norm=dk_ctables.cm_qpf_nws(atime=24)
cmap.set_under(color=[0,0,0,0],alpha=0.0)
plots['rain'] = ax.pcolormesh(
x,y,znan, norm=norm,
cmap=cmap, zorder=1,transform=datacrs,alpha=0.5)
#additional information
plt.title('['+rain.attrs['model']+'] '+
str(int(rain.coords['forecast_period'].values[0]))+'至'+str(int(rain.coords['forecast_period'].values[-1]))+'时效累积降水预报',
loc='left', fontsize=30)
ax.add_feature(cfeature.OCEAN)
utl.add_china_map_2cartopy_public(
ax, name='coastline', edgecolor='gray', lw=0.8, zorder=3,alpha=0.5)
if add_china:
utl.add_china_map_2cartopy_public(
ax, name='province', edgecolor='gray', lw=0.5, zorder=3)
utl.add_china_map_2cartopy_public(
ax, name='nation', edgecolor='black', lw=0.8, zorder=3)
utl.add_china_map_2cartopy_public(
ax, name='river', edgecolor='#74b9ff', lw=0.8, zorder=3,alpha=0.5)
# grid lines
gl = ax.gridlines(
crs=datacrs, linewidth=2, color='gray', alpha=0.5, linestyle='--', zorder=1)
gl.xlocator = mpl.ticker.FixedLocator(np.arange(0, 360, 15))
gl.ylocator = mpl.ticker.FixedLocator(np.arange(-90, 90, 15))
utl.add_cartopy_background(ax,name='RD')
l, b, w, h = ax.get_position().bounds
#forecast information
initTime = pd.to_datetime(
str(rain.coords['forecast_reference_time'].values)).replace(tzinfo=None).to_pydatetime()
#发布时间
if(sys.platform[0:3] == 'lin'):
locale.setlocale(locale.LC_CTYPE, 'zh_CN.utf8')
if(sys.platform[0:3] == 'win'):
locale.setlocale(locale.LC_CTYPE, 'chinese')
# add color bar
cax=plt.axes([l,b-0.04,w,.02])
cb = plt.colorbar(plots['rain'], cax=cax, orientation='horizontal')
cb.ax.tick_params(labelsize='x-large')
cb.set_label('Precipitation (mm)',size=20)
fcst_time=initTime+timedelta(hours=rain.coords['forecast_period'].values[0])
bax=plt.axes([l,b+h-0.07,.45,.07])
bax.set_yticks([])
bax.set_xticks([])
bax.axis([0, 10, 0, 10])
fcst_time1=initTime+timedelta(hours=rain.coords['forecast_period'].values[0]-6)
bax.text(2.5, 7.5,'起始时间: '+fcst_time1.strftime("%Y年%m月%d日%H时"),size=15)
bax.text(2.5, 0.5,'www.nmc.cn',size=15)
valid_fhour=bax.text(2.5, 5,'截至时间: ',size=15)
txt_fhour=bax.text(2.5, 2.5,'预报时效: ',size=15)
utl.add_logo_extra_in_axes(pos=[l-0.0,b+h-0.085,.1,.1],which='nmc', size='Xlarge')
# add south China sea
if south_China_sea:
utl.add_south_China_sea(pos=[l+w-0.091,b,.1,.2])
small_city=False
if(map_extent2[1]-map_extent2[0] < 25):
small_city=True
if city:
utl.add_city_on_map(ax,map_extent=map_extent2,transform=datacrs,zorder=2,size=13,small_city=small_city)
fcst_time2=initTime+timedelta(hours=rain.coords['forecast_period'].values[-1])
valid_fhour.set_text('截至时间: '+fcst_time2.strftime("%Y年%m月%d日%H时"))
txt_fhour.set_text('预报时效: '+str(int(rain.coords['forecast_period'].values[-1]))+'小时')
ax.pcolormesh(
x,y,np.squeeze(z[-1,:,:]), norm=norm,
cmap=cmap, zorder=1,transform=datacrs,alpha=0.5)
# plt.draw()
plt.savefig(output_dir+
'起报时间_'+initTime.strftime("%Y年%m月%d日%H时")+
'预报时效_'+str(int(rain.coords['forecast_period'].values[-1]))+'小时_'+rain.attrs['model']+'.png', dpi=200,bbox_inches='tight')
def draw_gh_uv_wvfl(gh=None,
uv=None,
wvfl=None,
map_extent=(50, 150, 0, 65),
regrid_shape=20,
add_china=True,
city=True,
south_China_sea=True,
output_dir=None,
Global=False):
plt.rcParams['font.sans-serif'] = ['SimHei'] # 步骤一（替换sans-serif字体）
plt.rcParams['axes.unicode_minus'] = False # 步骤二（解决坐标轴负数的负号显示问题）
# draw figure
plt.figure(figsize=(16, 9))
# set data projection
if (Global == True):
plotcrs = ccrs.Robinson(central_longitude=115.)
else:
plotcrs = ccrs.AlbersEqualArea(
central_latitude=(map_extent[2] + map_extent[3]) / 2.,
central_longitude=(map_extent[0] + map_extent[1]) / 2.,
standard_parallels=[30., 60.])
ax = plt.axes([0.01, 0.1, .98, .84], projection=plotcrs)
plt.title('[' + gh['model'] + '] ' + gh['lev'] + 'hPa 位势高度场, ' +
uv['lev'] + 'hPa 风场, ' + wvfl['lev'] + 'hPa 水汽通量',
loc='left',
fontsize=30)
datacrs = ccrs.PlateCarree()
#adapt to the map ratio
map_extent2 = utl.adjust_map_ratio(ax,
map_extent=map_extent,
datacrs=datacrs)
#adapt to the map ratio
ax.add_feature(cfeature.OCEAN)
utl.add_china_map_2cartopy_public(ax,
name='coastline',
edgecolor='gray',
lw=0.8,
zorder=105,
alpha=0.5)
if add_china:
utl.add_china_map_2cartopy_public(ax,
name='province',
edgecolor='gray',
lw=0.5,
zorder=105)
utl.add_china_map_2cartopy_public(ax,
name='nation',
edgecolor='black',
lw=0.8,
zorder=105)
utl.add_china_map_2cartopy_public(ax,
name='river',
edgecolor='#74b9ff',
lw=0.8,
zorder=105,
alpha=0.5)
# define return plots
plots = {}
# draw mean sea level pressure
if wvfl is not None:
x, y = np.meshgrid(wvfl['lon'], wvfl['lat'])
z = np.squeeze(wvfl['data']) / 10.
#pos=[0, 2, 4, 6, 8, 10, 12, 14, 16, 20, 22]
idx_nan = np.where(z < 5)
z[idx_nan] = np.nan
cmap, norm = gy_ctables.wvfl_ctable()
cmap.set_under(color=[0, 0, 0, 0], alpha=0.0)
plots['wvfl'] = ax.pcolormesh(x,
y,
z,
cmap=cmap,
norm=norm,
zorder=100,
transform=datacrs,
alpha=0.5)
# draw -hPa wind bards
if uv is not None:
x, y = np.meshgrid(uv['lon'], uv['lat'])
u = np.squeeze(uv['udata']) * 2.5
v = np.squeeze(uv['vdata']) * 2.5
plots['uv'] = ax.barbs(x,
y,
u,
v,
length=6,
regrid_shape=regrid_shape,
transform=datacrs,
fill_empty=False,
sizes=dict(emptybarb=0.05),
zorder=110)
# draw -hPa geopotential height
if gh is not None:
x, y = np.meshgrid(gh['lon'], gh['lat'])
clevs_gh = np.append(
np.append(
np.arange(0, 480, 4),
np.append(np.arange(480, 584, 8), np.arange(580, 604, 4))),
np.arange(604, 2000, 8))
plots['gh'] = ax.contour(x,
y,
np.squeeze(gh['data']),
clevs_gh,
colors='black',
linewidths=2,
transform=datacrs,
zorder=110)
plt.clabel(plots['gh'],
inline=1,
fontsize=20,
fmt='%.0f',
colors='black')
# grid lines
gl = ax.gridlines(crs=datacrs,
linewidth=2,
color='gray',
alpha=0.5,
linestyle='--',
zorder=40)
gl.xlocator = mpl.ticker.FixedLocator(np.arange(0, 360, 15))
gl.ylocator = mpl.ticker.FixedLocator(np.arange(-90, 90, 15))
#http://earthpy.org/cartopy_backgroung.html
#C:\ProgramData\Anaconda3\Lib\site-packages\cartopy\data\raster\natural_earth
ax.background_img(name='RD', resolution='high')
#forecast information
bax = plt.axes([0.01, 0.835, .25, .1], facecolor='#FFFFFFCC')
bax.set_yticks([])
bax.set_xticks([])
bax.axis([0, 10, 0, 10])
initial_time = pd.to_datetime(str(
gh['init_time'])).replace(tzinfo=None).to_pydatetime()
fcst_time = initial_time + timedelta(hours=gh['fhour'])
#发布时间
if (sys.platform[0:3] == 'lin'):
locale.setlocale(locale.LC_CTYPE, 'zh_CN')
if (sys.platform[0:3] == 'win'):
locale.setlocale(locale.LC_CTYPE, 'chinese')
plt.text(2.5,
7.5,
'起报时间: ' + initial_time.strftime("%Y年%m月%d日%H时"),
size=15)
plt.text(2.5, 5, '预报时间: ' + fcst_time.strftime("%Y年%m月%d日%H时"), size=15)
plt.text(2.5, 2.5, '预报时效: ' + str(gh['fhour']) + '小时', size=15)
plt.text(2.5, 0.5, 'www.nmc.cn', size=15)
# add color bar
if (wvfl != None):
cax = plt.axes([0.11, 0.06, .86, .02])
cb = plt.colorbar(plots['wvfl'], cax=cax, orientation='horizontal')
cb.ax.tick_params(labelsize='x-large')
cb.set_label('Water Vapor Flux (0.1g/(cm*hPa*s)', size=20)
# cb.set_ticks([0,4,8,12,16,20,24])
# cb.set_ticklabels(['0','4','8','12','16','20','24'])
# add south China sea
if south_China_sea:
utl.add_south_China_sea(pos=[0.85, 0.13, .1, .2])
small_city = False
if (map_extent2[1] - map_extent2[0] < 25):
small_city = True
if city:
utl.add_city_on_map(ax,
map_extent=map_extent2,
transform=datacrs,
zorder=110,
size=13,
small_city=small_city)
utl.add_logo_extra_in_axes(pos=[-0.01, 0.835, .1, .1],
which='nmc',
size='Xlarge')
# show figure
if (output_dir != None):
plt.savefig(output_dir + '位势高度场_风场_水汽通量_预报_' + '起报时间_' +
initial_time.strftime("%Y年%m月%d日%H时") + '预报时效_' +
str(gh['fhour']) + '小时' + '.png',
dpi=200)
if (output_dir == None):
plt.show()
def draw_isentropic_uv(isentrh=None, isentuv=None, isentprs=None,
map_extent=(50, 150, 0, 65),
regrid_shape=20,
add_china=True,city=False,south_China_sea=True,
output_dir=None,Global=False):
plt.rcParams['font.sans-serif'] = ['SimHei'] # 步骤一（替换sans-serif字体）
plt.rcParams['axes.unicode_minus'] = False # 步骤二（解决坐标轴负数的负号显示问题）
# draw figure
plt.figure(figsize=(16,9))
# set data projection
if(Global == True):
plotcrs = ccrs.Robinson(central_longitude=115.)
else:
plotcrs = ccrs.AlbersEqualArea(central_latitude=(map_extent[2]+map_extent[3])/2.,
central_longitude=(map_extent[0]+map_extent[1])/2., standard_parallels=[30., 60.])
ax = plt.axes([0.01,0.1,.98,.84], projection=plotcrs)
plt.title('['+isentrh['model']+'] '+
isentrh['lev']+'等熵面 风场 相对湿度 气压',
loc='left', fontsize=30)
datacrs = ccrs.PlateCarree()
#adapt to the map ratio
map_extent2=utl.adjust_map_ratio(ax,map_extent=map_extent,datacrs=datacrs)
#adapt to the map ratio
ax.add_feature(cfeature.OCEAN)
utl.add_china_map_2cartopy_public(
ax, name='coastline', edgecolor='gray', lw=0.8, zorder=105,alpha=0.5)
if add_china:
utl.add_china_map_2cartopy_public(
ax, name='province', edgecolor='gray', lw=0.5, zorder=105)
utl.add_china_map_2cartopy_public(
ax, name='nation', edgecolor='black', lw=0.8, zorder=105)
utl.add_china_map_2cartopy_public(
ax, name='river', edgecolor='#74b9ff', lw=0.8, zorder=105,alpha=0.5)
# define return plots
plots = {}
# draw mean sea level pressure
if isentrh is not None:
x, y = np.meshgrid(isentrh['lon'], isentrh['lat'])
z=np.squeeze(isentrh['data'])
clevs_rh= range(10, 106, 5)
plots['isentrh'] = ax.contourf(
x, y, z,clevs_rh,cmap=plt.cm.gist_earth_r,
transform=datacrs,alpha=0.5,extend='both')
# draw -hPa wind bards
if isentuv is not None:
x, y = np.meshgrid(isentuv['lon'], isentuv['lat'])
u = np.squeeze(isentuv['isentu']) * 2.5
v = np.squeeze(isentuv['isentv']) * 2.5
plots['uv'] = ax.barbs(
x, y, u, v, length=6, regrid_shape=regrid_shape,
transform=datacrs, fill_empty=False, sizes=dict(emptybarb=0.05),
zorder=20)
# draw -hPa geopotential height
if isentprs is not None:
x, y = np.meshgrid(isentprs['lon'], isentprs['lat'])
clevs_isentprs = np.arange(0, 1000, 25)
z=gaussian_filter(np.squeeze(isentprs['data']),5)
plots['isentprs'] = ax.contour(
x, y, z , clevs_isentprs, colors='black',
linewidths=2, transform=datacrs, zorder=30)
plt.clabel(plots['isentprs'], inline=1, fontsize=20, fmt='%.0f',colors='black')
# grid lines
gl = ax.gridlines(
crs=datacrs, linewidth=2, color='gray', alpha=0.5, linestyle='--', zorder=40)
gl.xlocator = mpl.ticker.FixedLocator(np.arange(0, 360, 15))
gl.ylocator = mpl.ticker.FixedLocator(np.arange(-90, 90, 15))
#http://earthpy.org/cartopy_backgroung.html
#C:\ProgramData\Anaconda3\Lib\site-packages\cartopy\data\raster\natural_earth
ax.background_img(name='RD', resolution='high')
#forecast information
bax=plt.axes([0.01,0.835,.25,.1],facecolor='#FFFFFFCC')
bax.set_yticks([])
bax.set_xticks([])
bax.axis([0, 10, 0, 10])
initTime = pd.to_datetime(
str(isentrh['init_time'])).replace(tzinfo=None).to_pydatetime()
fcst_time=initTime+timedelta(hours=isentrh['fhour'])
#发布时间
if(sys.platform[0:3] == 'lin'):
locale.setlocale(locale.LC_CTYPE, 'zh_CN.utf8')
if(sys.platform[0:3] == 'win'):
locale.setlocale(locale.LC_CTYPE, 'chinese')
plt.text(2.5, 7.5,'起报时间: '+initTime.strftime("%Y年%m月%d日%H时"),size=15)
plt.text(2.5, 5,'预报时间: '+fcst_time.strftime("%Y年%m月%d日%H时"),size=15)
plt.text(2.5, 2.5,'预报时效: '+str(isentrh['fhour'])+'小时',size=15)
plt.text(2.5, 0.5,'www.nmc.cn',size=15)
# add color bar
if(isentrh != None):
cax=plt.axes([0.11,0.06,.86,.02])
cb = plt.colorbar(plots['isentrh'], cax=cax, orientation='horizontal',
ticks=clevs_rh[:],
extend='both',extendrect=False)
cb.ax.tick_params(labelsize='x-large')
cb.set_label('Relative Humidity (%)',size=20)
# add south China sea
if south_China_sea:
utl.add_south_China_sea(pos=[0.85,0.13,.1,.2])
small_city=False
if(map_extent2[1]-map_extent2[0] < 25):
small_city=True
if city:
utl.add_city_on_map(ax,map_extent=map_extent2,transform=datacrs,zorder=110,size=13,small_city=small_city)
utl.add_logo_extra_in_axes(pos=[-0.01,0.835,.1,.1],which='nmc', size='Xlarge')
# show figure
if(output_dir != None):
plt.savefig(output_dir+isentrh['lev']+'等熵面分析_相对湿度_风场_气压_预报_'+
'起报时间_'+initTime.strftime("%Y年%m月%d日%H时")+
'预报时效_'+str(isentrh['fhour'])+'小时'+'.png', dpi=200)
if(output_dir == None):
plt.show()
def OBS_Sounding_GeopotentialHeight(IR=None,
Sounding=None,
HGT=None,
map_extent=None,
city=True,
south_China_sea=True,
output_dir=None,
Channel='C009'):
#if(sys.platform[0:3] == 'win'):
plt.rcParams['font.sans-serif'] = ['SimHei'] # 步骤一（替换sans-serif字体）
plt.rcParams['axes.unicode_minus'] = False # 步骤二（解决坐标轴负数的负号显示问题）
# draw figure
plt.figure(figsize=(16, 9))
plotcrs = ccrs.AlbersEqualArea(
central_latitude=(map_extent[2] + map_extent[3]) / 2.,
central_longitude=(map_extent[0] + map_extent[1]) / 2.,
standard_parallels=[30., 60.])
# draw main figure
ax = plt.axes([0.01, 0.1, .98, .84], projection=plotcrs)
datacrs = ccrs.PlateCarree()
map_extent2 = adjust_map_ratio(ax, map_extent=map_extent, datacrs=datacrs)
# plot map background
ax.add_feature(cfeature.OCEAN)
add_china_map_2cartopy_public(ax,
name='coastline',
edgecolor='gray',
lw=0.8,
zorder=2,
alpha=0.5)
add_china_map_2cartopy_public(ax,
name='nation',
edgecolor='black',
lw=0.8,
zorder=2)
add_china_map_2cartopy_public(ax,
name='river',
edgecolor='#74b9ff',
lw=0.8,
zorder=2,
alpha=0.5)
# define return plots
plots = {}
# draw IR
if IR is not None:
x, y = np.meshgrid(IR['lon'], IR['lat'])
z = np.squeeze(IR['image'])
if (Channel == 'C009'):
cmap = dk_ctables.cm_wv_enhancement()
norm, cmap = colortables.get_with_range('WVCIMSS', 160, 300)
title_name = '水汽图像'
plots['IR'] = ax.pcolormesh(x,
y,
z,
norm=norm,
cmap=cmap,
zorder=1,
transform=datacrs)
if (Channel == 'C012'):
cmap = dk_ctables.cm_ir_enhancement1()
title_name = '红外(10.8微米)'
plots['IR'] = ax.pcolormesh(x - 10.,
y,
z,
vmin=105.,
vmax=335.0,
cmap=cmap,
zorder=1,
transform=datacrs)
plt.title('FY4A' + title_name + '观测' + ' 探空观测 高度场',
loc='left',
fontsize=30)
if (Sounding is not None):
x = np.squeeze(Sounding['lon'].values)
y = np.squeeze(Sounding['lat'].values)
idx_vld = np.where((Sounding['Wind_angle'].values != np.nan)
& (Sounding['Wind_speed'].values != np.nan))
u, v = utl.wind2UV(Winddir=Sounding['Wind_angle'].values[idx_vld],
Windsp=Sounding['Wind_speed'].values[idx_vld])
idx_null = np.where((u > 100) | (v > 100))
u[idx_null] = np.nan
v[idx_null] = np.nan
c_barb = {'C009': 'black', 'C012': 'white'}
plots['uv'] = ax.barbs(x,
y,
u,
v,
transform=datacrs,
fill_empty=False,
sizes=dict(emptybarb=0.0),
barb_increments={
'half': 2,
'full': 4,
'flag': 20
},
zorder=2,
color=c_barb[Channel],
alpha=0.7,
lw=1.5,
length=7)
# draw mean sea level pressure
if (HGT is not None):
x, y = np.meshgrid(HGT['lon'], HGT['lat'])
clevs = np.append(np.arange(480, 584, 8), np.arange(580, 604, 4))
plots_HGT = ax.contour(x,
y,
ndimage.gaussian_filter(np.squeeze(
HGT['data']),
sigma=1,
order=0),
levels=clevs,
colors='black',
alpha=1,
zorder=3,
transform=datacrs,
linewidths=3)
ax.clabel(plots_HGT, inline=1, fontsize=20, fmt='%.0f')
ax.contour(x,
y,
ndimage.gaussian_filter(np.squeeze(HGT['data']),
sigma=1,
order=0),
levels=[588],
colors='black',
alpha=1,
zorder=3,
transform=datacrs,
linewidths=6)
#if city:
gl = ax.gridlines(crs=datacrs,
linewidth=1,
color='gray',
alpha=0.5,
linestyle='--',
zorder=40)
gl.xlocator = mpl.ticker.FixedLocator(np.arange(0, 360, 15))
gl.ylocator = mpl.ticker.FixedLocator(np.arange(-90, 90, 15))
#utl.add_cartopy_background(ax,name='RD')
l, b, w, h = ax.get_position().bounds
if (sys.platform[0:3] == 'lin'):
locale.setlocale(locale.LC_CTYPE, 'zh_CN.utf8')
if (sys.platform[0:3] == 'win'):
locale.setlocale(locale.LC_CTYPE, 'chinese')
IR_time = pd.to_datetime(
IR.coords['time'].values[0]).to_pydatetime() + timedelta(hours=8)
Sounding_time = Sounding['time'][0].to_pydatetime()
HGT_initTime = pd.to_datetime(
str(HGT.coords['forecast_reference_time'].values)).replace(
tzinfo=None).to_pydatetime()
HGT_time = HGT_initTime + timedelta(
hours=HGT.coords['forecast_period'].values[0])
#logo
bax = plt.axes([l, b + h - 0.1, .25, .1], facecolor='#FFFFFFCC')
bax.set_yticks([])
bax.set_xticks([])
bax.axis([0, 10, 0, 10])
add_logo_extra_in_axes(pos=[l - 0.02, b + h - 0.1, .1, .1],
which='nmc',
size='Xlarge')
bax.text(2.5,
7.5,
'卫星图像观测时间: ' + IR_time.strftime("%Y年%m月%d日%H时"),
size=12)
bax.text(2.5,
5,
'探空观测时间: ' + Sounding_time.strftime("%Y年%m月%d日%H时"),
size=12)
bax.text(2.5, 2.5, '高度场时间: ' + HGT_time.strftime("%Y年%m月%d日%H时"), size=12)
bax.text(2.5, 0.5, 'www.nmc.cn', size=12)
initial_time = pd.to_datetime(str(
IR['time'].values[0])).replace(tzinfo=None).to_pydatetime()
#发布时间
if (sys.platform[0:3] == 'lin'):
locale.setlocale(locale.LC_CTYPE, 'zh_CN')
if (sys.platform[0:3] == 'win'):
locale.setlocale(locale.LC_CTYPE, 'chinese')
# add color bar
cax = plt.axes([l, b - 0.04, w, .02])
cb = plt.colorbar(plots['IR'],
cax=cax,
orientation='horizontal',
extend='both',
extendrect=False)
cb.ax.tick_params(labelsize='x-large')
cb.set_label('(K)', size=10)
# add south China sea
if south_China_sea:
add_south_China_sea(pos=[l + w - 0.091, b, .1, .2])
small_city = False
if (map_extent2[1] - map_extent2[0] < 25):
small_city = True
if city:
add_city_on_map(ax,
map_extent=map_extent2,
transform=datacrs,
zorder=2,
size=16,
small_city=small_city)
# show figure
if (output_dir != None):
plt.savefig(
output_dir + '卫星' +
#'_'+initial_time.strftime("%Y年%m月%d日%H时")+
'卫星观测时间_' +
pd.to_datetime(IR['time'].values[0]).strftime("%Y年%m月%d日%H时") +
'.png',
dpi=200,
bbox_inches='tight')
plt.close()
if (output_dir == None):
plt.show()
def draw_gh_uv_VVEL(gh=None,
uv=None,
VVEL=None,
map_extent=(50, 150, 0, 65),
regrid_shape=20,
add_china=True,
city=True,
south_China_sea=True,
output_dir=None,
Global=False):
plt.rcParams['font.sans-serif'] = ['SimHei'] # 步骤一（替换sans-serif字体）
plt.rcParams['axes.unicode_minus'] = False # 步骤二（解决坐标轴负数的负号显示问题）
# draw figure
plt.figure(figsize=(16, 9))
# set data projection
if (Global == True):
plotcrs = ccrs.Robinson(central_longitude=115.)
else:
plotcrs = ccrs.AlbersEqualArea(
central_latitude=(map_extent[2] + map_extent[3]) / 2.,
central_longitude=(map_extent[0] + map_extent[1]) / 2.,
standard_parallels=[30., 60.])
ax = plt.axes([0.01, 0.1, .98, .84], projection=plotcrs)
plt.title('[' + gh.attrs['model'] + '] ' +
str(int(gh['level'].values[0])) + 'hPa 位势高度场, ' +
str(int(uv['level'].values[0])) + 'hPa 风场和垂直气压速度',
loc='left',
fontsize=30)
datacrs = ccrs.PlateCarree()
#adapt to the map ratio
map_extent2 = utl.adjust_map_ratio(ax,
map_extent=map_extent,
datacrs=datacrs)
#adapt to the map ratio
ax.add_feature(cfeature.OCEAN)
utl.add_china_map_2cartopy_public(ax,
name='coastline',
edgecolor='gray',
lw=0.8,
zorder=5,
alpha=0.5)
if add_china:
utl.add_china_map_2cartopy_public(ax,
name='province',
edgecolor='gray',
lw=0.5,
zorder=5)
utl.add_china_map_2cartopy_public(ax,
name='nation',
edgecolor='black',
lw=0.8,
zorder=5)
utl.add_china_map_2cartopy_public(ax,
name='river',
edgecolor='#74b9ff',
lw=0.8,
zorder=5,
alpha=0.5)
# define return plots
plots = {}
# draw mean sea level pressure
if VVEL is not None:
x, y = np.meshgrid(VVEL['lon'], VVEL['lat'])
clevs_VVEL = [0.1, 4, 13, 25, 60, 120]
clevs_VVEL = [
-30, -20, -10, -5, -2.5, -1, -0.5, 0.5, 1, 2.5, 5, 10, 20, 30
]
z = np.squeeze(VVEL['data'] / 10.)
cmap, norm = dk_ctables.cm_vertical_velocity_nws(pos=clevs_VVEL)
#cmap.set_under(color=[0,0,0,0],alpha=0.0)
plots['VVEL'] = ax.pcolormesh(x,
y,
z,
norm=norm,
cmap=cmap,
zorder=1,
transform=datacrs,
alpha=0.5)
# draw -hPa wind bards
if uv is not None:
x, y = np.meshgrid(uv['lon'], uv['lat'])
u = np.squeeze(uv['u']) * 2.5
v = np.squeeze(uv['v']) * 2.5
plots['uv'] = ax.barbs(x,
y,
u.values,
v.values,
length=6,
regrid_shape=regrid_shape,
transform=datacrs,
fill_empty=False,
sizes=dict(emptybarb=0.05),
zorder=2)
# draw -hPa geopotential height
if gh is not None:
x, y = np.meshgrid(gh['lon'], gh['lat'])
clevs_gh = np.append(
np.append(
np.arange(0, 480, 4),
np.append(np.arange(480, 584, 8), np.arange(580, 604, 4))),
np.arange(604, 2000, 8))
plots['gh'] = ax.contour(x,
y,
np.squeeze(gh['data']),
clevs_gh,
colors='black',
linewidths=2,
transform=datacrs,
zorder=3)
plt.clabel(plots['gh'],
inline=1,
fontsize=20,
fmt='%.0f',
colors='black')
# grid lines
gl = ax.gridlines(crs=datacrs,
linewidth=2,
color='gray',
alpha=0.5,
linestyle='--',
zorder=4)
gl.xlocator = mpl.ticker.FixedLocator(np.arange(0, 360, 15))
gl.ylocator = mpl.ticker.FixedLocator(np.arange(-90, 90, 15))
utl.add_cartopy_background(ax, name='RD')
#forecast information
l, b, w, h = ax.get_position().bounds
#forecast information
bax = plt.axes([l, b + h - 0.1, .25, .1], facecolor='#FFFFFFCC')
bax.set_yticks([])
bax.set_xticks([])
bax.axis([0, 10, 0, 10])
initTime = pd.to_datetime(str(
gh.coords['forecast_reference_time'].values)).replace(
tzinfo=None).to_pydatetime()
fcst_time = initTime + timedelta(
hours=gh.coords['forecast_period'].values[0])
#发布时间
if (sys.platform[0:3] == 'lin'):
locale.setlocale(locale.LC_CTYPE, 'zh_CN.utf8')
if (sys.platform[0:3] == 'win'):
locale.setlocale(locale.LC_CTYPE, 'chinese')
plt.text(2.5, 7.5, '起报时间: ' + initTime.strftime("%Y年%m月%d日%H时"), size=15)
plt.text(2.5, 5, '预报时间: ' + fcst_time.strftime("%Y年%m月%d日%H时"), size=15)
plt.text(2.5,
2.5,
'预报时效: ' + str(int(gh.coords['forecast_period'].values[0])) +
'小时',
size=15)
plt.text(2.5, 0.5, 'www.nmc.cn', size=15)
# add color bar
if (VVEL != None):
cax = plt.axes([l, b - 0.04, w, .02])
cb = plt.colorbar(plots['VVEL'],
cax=cax,
orientation='horizontal',
ticks=clevs_VVEL[:],
extend='max',
extendrect=False)
cb.ax.tick_params(labelsize='x-large')
cb.set_label('Vertical Velocity (0.1Pa/s)', size=20)
# add south China sea
if south_China_sea:
utl.add_south_China_sea(pos=[l + w - 0.091, b, .1, .2])
small_city = False
if (map_extent2[1] - map_extent2[0] < 25):
small_city = True
if city:
utl.add_city_on_map(ax,
map_extent=map_extent2,
transform=datacrs,
zorder=110,
size=13,
small_city=small_city)
utl.add_logo_extra_in_axes(pos=[l - 0.02, b + h - 0.1, .1, .1],
which='nmc',
size='Xlarge')
# show figure
if (output_dir != None):
plt.savefig(output_dir + '高度场_风场_垂直气压速度_预报_' + '起报时间_' +
initTime.strftime("%Y年%m月%d日%H时") + '预报时效_' +
str(gh.coords['forecast_period'].values[0]) + '小时' +
'.png',
dpi=200,
bbox_inches='tight')
if (output_dir == None):
plt.show()
def draw_low_level_wind(uv=None,
map_extent=(50, 150, 0, 65),
regrid_shape=20,
add_china=True,
city=True,
south_China_sea=True,
output_dir=None,
Global=False):
# set font
plt.rcParams['font.sans-serif'] = ['SimHei'] # 步骤一（替换sans-serif字体）
plt.rcParams['axes.unicode_minus'] = False # 步骤二（解决坐标轴负数的负号显示问题）
# set figure
plt.figure(figsize=(16, 9))
if (Global == True):
plotcrs = ccrs.Robinson(central_longitude=115.)
else:
plotcrs = ccrs.AlbersEqualArea(
central_latitude=(map_extent[2] + map_extent[3]) / 2.,
central_longitude=(map_extent[0] + map_extent[1]) / 2.,
standard_parallels=[30., 60.])
datacrs = ccrs.PlateCarree()
ax = plt.axes([0.01, 0.1, .98, .84], projection=plotcrs)
# define return plots
plots = {}
# draw mean sea level pressure
if uv is not None:
x, y = np.meshgrid(uv['lon'], uv['lat'])
wsp = np.squeeze((uv['u'].values**2 + uv['v'].values**2)**0.5)
z = wsp
cmap = dk_ctables.cm_wind_speed_nws()
cmap.set_under(color=[0, 0, 0, 0], alpha=0.0)
plots['wsp'] = ax.pcolormesh(x,
y,
z,
cmap=cmap,
zorder=1,
transform=datacrs,
alpha=0.5)
# draw -hPa wind bards
if uv is not None:
x, y = np.meshgrid(uv['lon'], uv['lat'])
u = np.squeeze(uv['u'].values) * 2.5
v = np.squeeze(uv['v'].values) * 2.5
x, y = np.meshgrid(uv['lon'], uv['lat'])
lw = 5 * wsp / wsp.max()
plots['stream'] = ax.streamplot(x,
y,
u,
v,
density=2,
color='r',
linewidth=lw,
zorder=100,
transform=datacrs)
spd_rtio = 0.03
for i in range(0, len(uv['lon']) - 1, 10):
for j in range(0, len(uv['lat']) - 1, 10):
ax.arrow(x[j, i],
y[j, i],
u[j, i] * spd_rtio,
v[j, i] * spd_rtio,
color='black',
zorder=100,
transform=datacrs,
width=0.05)
plt.title('[' + uv.attrs['model'] + '] ' + uv.attrs['level'] + ' 风场, 风速',
loc='left',
fontsize=30)
#adapt to the map ratio
map_extent2 = utl.adjust_map_ratio(ax,
map_extent=map_extent,
datacrs=datacrs)
#adapt to the map ratio
ax.add_feature(cfeature.OCEAN)
utl.add_china_map_2cartopy_public(ax,
name='coastline',
edgecolor='gray',
lw=0.8,
zorder=2,
alpha=0.5)
if add_china:
utl.add_china_map_2cartopy_public(ax,
name='province',
edgecolor='gray',
lw=0.5,
zorder=2)
utl.add_china_map_2cartopy_public(ax,
name='nation',
edgecolor='black',
lw=0.8,
zorder=2)
utl.add_china_map_2cartopy_public(ax,
name='river',
edgecolor='#74b9ff',
lw=0.8,
zorder=2,
alpha=0.5)
# grid lines
gl = ax.gridlines(crs=datacrs,
linewidth=2,
color='gray',
alpha=0.5,
linestyle='--',
zorder=3)
gl.xlocator = mpl.ticker.FixedLocator(np.arange(0, 360, 15))
gl.ylocator = mpl.ticker.FixedLocator(np.arange(-90, 90, 15))
utl.add_cartopy_background(ax, name='RD')
l, b, w, h = ax.get_position().bounds
#forecast information
bax = plt.axes([l, b + h - 0.1, .25, .1], facecolor='#FFFFFFCC')
bax.set_yticks([])
bax.set_xticks([])
bax.axis([0, 10, 0, 10])
initTime = pd.to_datetime(str(
uv.coords['forecast_reference_time'].values)).replace(
tzinfo=None).to_pydatetime()
fcst_time = initTime + timedelta(
hours=uv.coords['forecast_period'].values[0])
#发布时间
if (sys.platform[0:3] == 'lin'):
locale.setlocale(locale.LC_CTYPE, 'zh_CN.utf8')
if (sys.platform[0:3] == 'win'):
locale.setlocale(locale.LC_CTYPE, 'chinese')
plt.text(2.5, 7.5, '起报时间: ' + initTime.strftime("%Y年%m月%d日%H时"), size=15)
plt.text(2.5, 5, '预报时间: ' + fcst_time.strftime("%Y年%m月%d日%H时"), size=15)
plt.text(2.5,
2.5,
'预报时效: ' + str(int(uv.coords['forecast_period'].values[0])) +
'小时',
size=15)
plt.text(2.5, 0.5, 'www.nmc.cn', size=15)
# add color bar
if (wsp is not None):
cax = plt.axes([l, b - 0.04, w, .02])
cb = plt.colorbar(plots['wsp'],
cax=cax,
orientation='horizontal',
extend='max',
extendrect=False)
cb.ax.tick_params(labelsize='x-large')
cb.set_label('(m/s)', size=20)
# add south China sea
if south_China_sea:
utl.add_south_China_sea(pos=[l + w - 0.091, b, .1, .2])
small_city = False
if (map_extent2[1] - map_extent2[0] < 25):
small_city = True
if city:
utl.add_city_on_map(ax,
map_extent=map_extent2,
transform=datacrs,
zorder=4,
size=13,
small_city=small_city)
utl.add_logo_extra_in_axes(pos=[l - 0.02, b + h - 0.1, .1, .1],
which='nmc',
size='Xlarge')
# show figure
if (output_dir != None):
plt.savefig(output_dir + '低层风_预报_' + '起报时间_' +
initTime.strftime("%Y年%m月%d日%H时") + '预报时效_' +
str(uv.coords['forecast_period'].values[0]) + '小时' +
'.png',
dpi=200,
bbox_inches='tight')
plt.close()
if (output_dir == None):
plt.show()
