-
Notifications
You must be signed in to change notification settings - Fork 0
/
plt_ufo_omb_uv.py
120 lines (98 loc) · 3.36 KB
/
plt_ufo_omb_uv.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
import numpy as np
import sys
import statistics
import math
import matplotlib.pyplot as plt
from netCDF4 import Dataset
#from datetime import datetime
#import matplotlib.colors as colors
#import cartopy.crs as ccrs
#import cartopy.feature as cfeature
def plt_ufo_t(filename,OBSTYPE,VarName):
thisobstype=OBSTYPE
thisvarname=VarName
gsihofXBc=thisvarname+'@GsiHofXBc'
gsihofX =thisvarname+'@GsiHofX'
ufohofX =thisvarname+'@hofx'
f=Dataset(filename, mode='r')
gsi_observer_withqc=f.variables[gsihofXBc][:]
gsi_observer_noqc =f.variables[gsihofX][:]
ufo =f.variables[ufohofX][:]
geopotential_height=f.variables['geopotential_height@MetaData'][:]
# gsi_observer_withqc=f.variables['radial_velocity@GsiHofXBc'][:]
f.close()
plt.rcParams.update({'font.size': 18})
# plt.rcParams.update({'line.linewidth': 8})
#=========================
fig = plt.figure(figsize=(8.0,7.5))
ax=fig.add_subplot(111)
#plt.plot(gsi_observer,ufo, color='black',label="obs" , linewidth=2, marker='o')
# plt.scatter(gsi_observer_withqc,ufo, color='blue',label="rw", marker='o', s=3)
plt.scatter(gsi_observer_noqc,ufo, color='r',label="tsen", marker='o', s=3)
#box = ax.get_position()
#ax.set_position([box.x0, box.y0, box.width * 0.8, box.height])
# Put a legend to the right of the current axis
# ax.legend(loc='center left', bbox_to_anchor=(1, 0.5))
plt.xlabel('gsi')
plt.ylabel('ufo')
plt.title(thisobstype+':gsi and ufo hofx scatter')
figname='ufo_'+thisobstype+'_stage1.png'
plt.savefig(figname,bbox_inches='tight',dpi=100)
#=========================
#=========================
diff=gsi_observer_noqc
diff=diff - ufo
print(diff)
rms=float(0)
for x in diff:
rms=rms+x*x
rms=math.sqrt(rms/len(diff))
print("rms=",rms)
diff=diff*1000.0
print(diff.max(),diff.min())
fig1 = plt.figure(figsize=(8.0,7.5))
ax=fig1.add_subplot(111)
plt.hist(diff,bins=50,range=(-0.10,0.10))
# plt.hist(diff,bins=50,range=(-2.00,2.00))
# plt.xlim([-0.15,0.15])
plt.xlabel('(gsi-ufo)*1000')
plt.title(thisobstype+':gsi and ufo diff histogram')
figname='ufo_'+thisobstype+'_stage1_hist.png'
plt.savefig(figname,bbox_inches='tight',dpi=100)
#=========================
#=========================
fig2 = plt.figure(figsize=(8.0,7.5))
ax=fig2.add_subplot(111)
plt.scatter(diff,geopotential_height, color='b',label="rw", marker='o', s=3)
# plt.xlim([-0.15,0.15])
plt.xlabel('(gsi-ufo)*1000')
plt.ylabel('geop-height')
plt.title(thisobstype+':gsi-ufo diff in vertical')
figname='ufo_'+thisobstype+'_stage1_vdiff_scatter.png'
plt.savefig(figname,bbox_inches='tight',dpi=100)
#=========================
#=====================================================================
#=====================================================================
if __name__ == '__main__':
print("start ploting")
#call subroutine plt_radar_ref
print("ploting gsi hofx v.s. ufo hofx")
fileame=sys.argv[1]
OBSTYPE=sys.argv[2]
VarName=sys.argv[3]
plt_ufo_t(fileame,OBSTYPE,VarName)
# print(statistics.stdev(diff))
#plt.xticks(np.arange(0,25,3))
#ymin1=imos[0:8].min()
#ymin2=gfs.min()
#ymin=min(ymin1,ymin2)
#ymin=ymin-1.0
#print ymin
#ymax1=imos[0:8].max()
#ymax2=gfs.max()
#ymax=max(ymax1,ymax2)
#ymax=ymax+1.0
#plt.ylim([ymin,ymax])
#plt.xlim([-25,25])
# plt.xticks(fontsize=15)
# plt.yticks(fontsize=15)