Plumb通量计算

凯门鳄

这个Plumb通量的xy方向是抄的摸鱼师兄在B站讲的，自己在他讲的基础上加的一个z方向，但是结果好像不对。有没有大佬能看看问题出在哪
a=6370000
omega=7.292e-5
g = 9.8
f = (2 * omega * np.sin((np.array(lat))*np.pi/180)).reshape((1,lat.shape[0],1))
# print(f.shape)
p_ = [1000.,  925.,  850.,  700.,  600.,  500.,  400.,  300.,  250.,  200.,
        150.,  100.,   70.,   50.,   30.,   20.,   10.]
p = np.array(p_)
p_p0=(p/1000).reshape((-1,1,1))
coslat = (np.cos(np.array(lat)*np.pi/180)).reshape((1,lat.shape[0],1))   #同样是增加一个维度计算
sin2lat = (np.sin(np.array(lat)*2*np.pi/180)).reshape((1,lat.shape[0],1))

dlon=(np.gradient(lon)*np.pi/180.0).reshape((1,1,lon.shape[0]))  #得到的结果转化为弧度  （1，x）   （x，144）
dlat=(np.gradient(lat)*np.pi/180.0).reshape((1,lat.shape[0],1))

dtzdlon = np.gradient(ta * za ,axis = 2)/dlon

zz = (-8000 * np.log(p/1000))
dt = (np.gradient(tam ,axis = 0))
dz = (np.gradient(zz)).reshape((-1,1,1))
S = dt/dz+0.286*tam/8000

fz_= p_p0*coslat*(f/S*(ta * va - dtzdlon/(sin2lat*2*omega*a)))     #(17,25,144)