用advection函数求垂直水汽平流

长征火箭兔

想通过metpy库的advection函数求整层的水平水汽平流和垂直水汽平流，公式如下 。我的计算水平水汽平流的代码如下

1. for k in range(lev.shape[0]):
   
2.       adv_vdq3[k] = mpcalc.advection(qa3[k] ,u = ua3[k] ,v = va3[k] ,dx = dx ,dy = dy)*1e5
   
3. 
   
4. total_adv_vdq3[t] = np.trapz(adv_vdq3[t,::-1,:,:],lev[::-1],axis=0)

复制代码

但是在计算垂直水汽平流的时候，如何计算dz呢，这是我写的关于dz的计算。

1. wa = w[:,:-1,:,:]  
   
2. wb = w[:,1:,:,:]   
   
3. dz = np.array(wb)-np.array(wa)

复制代码

下面是报错

1. axis -3 is out of bounds for array of dimension 2

复制代码

咣当当

我最近也遇到了这个问题，请问楼主解决了吗？如何计算垂直水汽平流呀？