def draw_point_fcst(t2m=None,
u10m=None,
v10m=None,
rn=None,
model=None,
output_dir=None,
points=None,
extra_info=None):
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')
initTime = pd.to_datetime(str(
t2m['forecast_reference_time'].values)).replace(
tzinfo=None).to_pydatetime()
# draw figure
fig = plt.figure(figsize=(16, 4.5))
ax_t2m = HostAxes(fig, [0.1, 0.28, .8, .62])
ax_rn = ParasiteAxes(ax_t2m, sharex=ax_t2m)
#其他信息
#append axes
ax_t2m.parasites.append(ax_rn)
#invisible right axis of ax
ax_t2m.axis['right'].set_visible(False)
ax_t2m.axis['right'].set_visible(False)
ax_rn.axis['right'].set_visible(True)
ax_rn.axis['right'].major_ticklabels.set_visible(True)
ax_rn.axis['right'].label.set_visible(True)
#set label for axis
ax_t2m.set_ylabel('温度($^\circ$C)', fontsize=100)
ax_rn.set_ylabel('降水(mm)', fontsize=100)
fig.add_axes(ax_t2m)
# draw main figure
#2米温度——————————————————————————————————————
if (model == '中央台指导'):
model = '智能网格'
utl.add_public_title_sta(title=model + '预报 ' + extra_info['point_name'] +
' [' + str(points['lon'][0]) + ',' +
str(points['lat'][0]) + ']',
initTime=initTime,
fontsize=21)
for ifhour in t2m['forecast_period'].values:
if (ifhour == t2m['forecast_period'].values[0]):
t2m_t = (initTime + timedelta(hours=ifhour))
else:
t2m_t = np.append(t2m_t, (initTime + timedelta(hours=ifhour)))
curve_t2m = ax_t2m.plot(t2m_t,
np.squeeze(t2m['data'].values),
c='#FF6600',
linewidth=3,
label='2m温度')
ax_t2m.set_xlim(t2m_t[0], t2m_t[-1])
ax_t2m.set_ylim(
math.floor(t2m['data'].values.min() / 5) * 5 - 2,
math.ceil(t2m['data'].values.max() / 5) * 5)
#降水——————————————————————————————————————
for ifhour in rn['forecast_period'].values:
if (ifhour == rn['forecast_period'].values[0]):
rn_t = (initTime + timedelta(hours=ifhour))
else:
rn_t = np.append(rn_t, (initTime + timedelta(hours=ifhour)))
mask = (rn['data'] < 999)
rn = rn['data'].where(mask)
ax_rn.bar(rn_t,
np.squeeze(rn.values),
width=0.1,
color='#00008B',
label=str(
int(rn['forecast_period'].values[1] -
rn['forecast_period'].values[0])) + '小时降水',
alpha=0.5)
#curve_rn=ax_rn.plot(rn_t, np.squeeze(rn['data'].values), c='#40C4FF',linewidth=3)
ax_rn.set_ylim(0, np.nanmax(np.append(10, np.squeeze(rn.values))))
###
xaxis_intaval = mpl.dates.HourLocator(byhour=(8, 20)) #单位是小时
ax_t2m.xaxis.set_major_locator(xaxis_intaval)
# add legend
ax_t2m.legend(fontsize=15, loc='upper right')
ax_t2m.tick_params(length=10)
ax_t2m.tick_params(axis='y', labelsize=100)
ax_t2m.set_xticklabels([' '])
miloc = mpl.dates.HourLocator(byhour=(8, 11, 14, 17, 20, 23, 2, 5)) #单位是小时
ax_t2m.xaxis.set_minor_locator(miloc)
yminorLocator = MultipleLocator(1) #将此y轴次刻度标签设置为1的倍数
ax_t2m.yaxis.set_minor_locator(yminorLocator)
ymajorLocator = MultipleLocator(5) #将此y轴次刻度标签设置为1的倍数
ax_t2m.yaxis.set_major_locator(ymajorLocator)
ax_t2m.grid(axis='x', which='minor', ls='--')
ax_t2m.axis['left'].label.set_fontsize(15)
ax_t2m.axis['left'].major_ticklabels.set_fontsize(15)
ax_rn.axis['right'].label.set_fontsize(15)
ax_rn.axis['right'].major_ticklabels.set_fontsize(15)
#10米风——————————————————————————————————————
ax_uv = plt.axes([0.1, 0.16, .8, .12])
for ifhour in u10m['forecast_period'].values:
if (ifhour == u10m['forecast_period'].values[0]):
uv_t = (initTime + timedelta(hours=ifhour))
else:
uv_t = np.append(uv_t, (initTime + timedelta(hours=ifhour)))
wsp = (u10m**2 + v10m**2)**0.5
#curve_uv=ax_uv.plot(uv_t, np.squeeze(wsp['data'].values), c='#696969',linewidth=3,label='10m风')
ax_uv.barbs(uv_t,
np.zeros(len(uv_t)),
np.squeeze(u10m['data'].values),
np.squeeze(v10m['data'].values),
fill_empty=True,
color='gray',
barb_increments={
'half': 2,
'full': 4,
'flag': 20
},
length=5.8,
linewidth=1.5,
zorder=100)
ax_uv.set_ylim(-1, 1)
ax_uv.set_xlim(uv_t[0], uv_t[-1])
#ax_uv.axis('off')
ax_uv.set_yticklabels([' '])
#logo
utl.add_logo_extra_in_axes(pos=[0.87, 0.00, .1, .1],
which='nmc',
size='Xlarge')
#开启自适应
xaxis_intaval = mpl.dates.HourLocator(byhour=(8, 20)) #单位是小时
ax_uv.xaxis.set_major_locator(xaxis_intaval)
ax_uv.tick_params(length=5, axis='x')
ax_uv.tick_params(length=0, axis='y')
miloc = mpl.dates.HourLocator(byhour=(8, 11, 14, 17, 20, 23, 2, 5)) #单位是小时
ax_uv.xaxis.set_minor_locator(miloc)
ax_uv.grid(axis='x', which='both', ls='--')
ax_uv.set_ylabel('10m风', fontsize=15)
xstklbls = mpl.dates.DateFormatter('%m月%d日%H时')
for label in ax_uv.get_xticklabels():
label.set_rotation(30)
label.set_horizontalalignment('center')
ax_uv.tick_params(axis='x', labelsize=15)
#出图——————————————————————————————————————————————————————————
if (output_dir != None):
isExists = os.path.exists(output_dir)
if not isExists:
os.makedirs(output_dir)
output_dir2 = output_dir + model + '_起报时间_' + initTime.strftime(
"%Y年%m月%d日%H时") + '/'
if (os.path.exists(output_dir2) == False):
os.makedirs(output_dir2)
plt.savefig(output_dir2 + model + '_' + extra_info['point_name'] +
'_' + extra_info['output_head_name'] +
initTime.strftime("%Y%m%d%H") + '00' +
extra_info['output_tail_name'] + '.jpg',
dpi=200,
bbox_inches='tight')
else:
plt.show()
def plot_line_(name, x, y1, y2, y3):
#fig = plt.figure(figsize=(7,5)) #figsize默认为4,4(图像尺寸),facecolor="blue"
fig = plt.figure(1) #figsize默认为4,4(图像尺寸),facecolor="blue"
ax = HostAxes(
fig, [0.15, 0.1, 0.7, 0.8
]) #用[left, bottom, weight, height]的方式定义axes，0 <= l,b,w,h <= 1
ax_ndvi = ParasiteAxes(ax, sharex=ax)
ax_t = ParasiteAxes(ax, sharex=ax)
ax_p = ParasiteAxes(ax, sharex=ax)
ax.parasites.append(ax_ndvi)
ax.parasites.append(ax_p)
ax.parasites.append(ax_t)
ax.set_ylim(0.41, 0.75)
ax.set_xlim(2004.5, 2015.5)
ax.axis['right'].set_visible(False)
ax.axis['top'].set_visible(False)
# plt.tick_params(top = 'off', right = 'off')
#ax_ndvi.axis['left1'] = new_axisline(loc='left', axes=ax_ndvi, offset=offset1)
#ax.axis['left'].set_axisline_style('->',size=2) #轴的形状色
ax.axis['left'].line.set_linewidth(3) #轴宽
ax.axis['bottom'].line.set_linewidth(3) #轴宽
offset1 = (20, 0)
offset2 = (60, 0)
ax_t.set_ylim(10, 30)
ax_p.set_ylim(500, 2500)
new_axisline = ax_t._grid_helper.new_fixed_axis # "_grid_helper"与"get_grid_helper()"等价，可以代替 #new_axisline = par2.get_grid_helper().new_fixed_axis # 用"get_grid_helper()"代替，结果一样，区别目前不清楚
new_axisline = ax_p._grid_helper.new_fixed_axis
ax_t.axis['right2'] = new_axisline(loc='right', axes=ax_t, offset=offset1)
#ax_t.axis['right2'].set_axisline_style('-|>',size=-1) #轴的形状色
ax_t.axis['right2'].line.set_linewidth(3) #轴宽
ax_t.axis['right2'].set_label(
'T') # ax_t.axis['right2'].set_axislabel_direction('-')
ax_p.axis['right3'] = new_axisline(loc='right', axes=ax_p, offset=offset2)
#ax_p.axis['right3'].set_axisline_style('->',size=-1) #轴的形状色
ax_p.axis['right3'].line.set_linewidth(3) #轴宽
ax_p.axis['right3'].set_label('P')
ax_t.yaxis.set_major_locator(MultipleLocator(2))
ax_p.yaxis.set_major_locator(MultipleLocator(500))
ax.xaxis.set_major_locator(
xmajorLocator
) #设置主刻度标签的位置,没有标签文本格式ax.xaxis.set_major_formatter(FormatStrFormatter('%5.1f'))
ax.yaxis.set_major_locator(ymajorLocator)
ax.xaxis.set_minor_locator(xminorLocator) #设置次刻度标签的位置,没有标签文本格式
ax.xaxis.set_minor_locator(xminorLocator)
#打开网格
ax_ndvi.grid(which='minor', color='black',
linestyle='--') #,lw=0.8, alpha=0.5) # grid setting) #绘制网格线
ax_t.grid(which='minor', color='black',
linestyle='--') #,lw=0.8, alpha=0.5) # grid setting) #绘制网格线
ax.tick_params(axis='both',
which='major',
direction='in',
width=2,
length=lwidth,
pad=tick_pad,
labelsize=tick_labelsz,
grid_linewidth=3)
ax.set_ylabel('NDVI', font_properties=getChineseFont(80))
#ax.set_ylabel('NDVI',fontproperties=getChineseFont(100))
ax_t.tick_params(axis='both',
which='both',
direction='in',
width=3,
length=100,
pad=18,
labelsize=100,
grid_linewidth=1)
ax_p.tick_params(axis='both',
which='both',
direction='in',
width=3,
length=100,
pad=18,
labelsize=100,
grid_linewidth=1)
fig.add_axes(ax)
ax.set_xlabel('Year')
p1, = ax.plot(x,
y1,
label="NDVI",
ls=linestyle,
lw=lwidth,
color=c1,
marker=".",
ms=16,
mfc=c1)
p2, = ax_t.plot(x,
y3,
label="Temperature",
ls=linestyle,
lw=lwidth,
color=c2,
marker="o",
ms=marksize,
mfc=c2)
p3, = ax_p.plot(x,
y2,
label="Precipitation",
ls=linestyle,
lw=lwidth,
color=c3,
marker="o",
ms=marksize,
mfc=c3)
#ax.set_xlabel('YEAR')
#ax.set_ylabel('NDVI')
plt.legend(loc=1,
bbox_to_anchor=(0.9, 1.2),
prop=getChineseFont(12),
ncol=1,
frameon=False,
mode='expend')
#plt.savefig(r'C:\Users\YeHui\Desktop\GGGGGGGGGGG.jpg')#, dpi=1000)
plt.show()
