python 计算 Rainfall Anomaly Index (RAI)

haysen

1. import numpy as np
   
2. #参考 https://github.com/lucasvenez/precintcon/blob/master/R/precintcon.rai.analysis.r
   
3. def rai(datanp):
   
4.         sort_datanp = datanp.copy()
   
5.         sort_datanp.sort()
   
6. 
   
7.         xb = sort_datanp[0:10].mean()
   
8.         mb = sort_datanp[-10:].mean()
   
9.         k = datanp.mean()
   
10. 
    
11.         for i in range(len(datanp)):
    
12.                 x = datanp[i]
    
13.                 if x < k:
    
14.                         datanp[i] = -3 * ((x - k) / (xb - k))
    
15.                 else:
    
16.                         datanp[i] = 3 * ((x - k) / (mb - k))
    
17.         return datanp
    
18. 
    
19. 
    
20. datanp = np.arange(200)
    
21. 
    
22. print(rai(datanp))
    
23. 
    
24.         

复制代码

chongzika

最近，我也注意到了这个R包