Python计算相关系数，一个是时间序列和格点的相关，一个是空间相关

洗雨

纯属瞎折腾，希望得到指点

1. # -*- coding: utf-8 -*-
   
2. """
   
3. Created on Wed Jan 13 16:56:43 2021
   
4. 
   
5. @author: gaoyong
   
6. """
   
7. import xarray as xr
   
8. import numpy as np
   
9. import pandas as pd
   
10. from scipy import stats
    
11. import calendar as cal
    
12. 
    
13. 
    
14. 
    
15. def time_series(filename,yearstart,monthstart,yearend,monthend):
    
16.     file2 = pd.read_csv(filename,sep='\s+')
    
17.     ser   = []
    
18.   
    
19.     for j in range(yearstart,yearend+1):
    
20.         for i in range(monthstart,monthend+1):
    
21.             imonth = cal.month_abbr[i]
    
22.             ser.append(file2[imonth][j])
    
23.     return ser
    
24. 
    
25. def grid_series_corr(ncfilename,ser):
    
26.     data = xr.open_dataset(ncfilename)
    
27.     ser  = np.array(ser)
    
28.     yl,ml,latl,lonl = data.pre.shape
    
29.     data_new        = np.array((data.pre.values).reshape(yl*ml,latl*lonl))
    
30.     latmin          = data.lat.min()
    
31.     latmax          = data.lat.max()
    
32.     lonmin          = data.lon.min()
    
33.     lonmax          = data.lon.max()
    
34.     lats            = np.linspace(latmax,latmin,latl)
    
35.     lons            = np.linspace(lonmin,lonmax,lonl)
    
36.     rr              = []
    
37.     for i in range(latl*lonl):
    
38.         r = np.corrcoef(np.array(data_new[:,i]),ser)[0,1]
    
39.         rr.append(r)
    
40.     lons,lats = np.meshgrid(lons,lats)
    
41.     rr        = np.array(rr).reshape(lonl,latl)
    
42.     return rr,lons,lats
    
43. 
    
44. def spacc_corr(data1,data2):
    
45.     '''data1和data2为2维数组,数据维度相同,还没写完，慎用'''
    
46.     data1 = np.array(data1)
    
47.     data2 = np.array(data2)
    
48.     x,y  = data1.shape
    
49.     co = []
    
50.     for i in range(y):
    
51.         r  = np.corrcoef(data1[:,i],data2[:,i])[0,1]
    
52.         co.append(r)
    
53.     co = np.array(co).reshape(x,y)
    
54.     return co
    
55. 
    
56. if __name__ == '__main__':
    
57.     file   = r'E:\tem_test\tw\lqx\pre.mon.1961-2018.nc'
    
58.     file2  = r'E:\tem_test\tw\lqx\AOI.txt'
    
59. 
    
60. 
    
61.     ser = time_series(file2,1961,1,2018,12)
    
62.     a,lons,lats   = grid_series_corr(file,ser)
    
63.     print(a)

复制代码

shower107

def spacc_corr(data1,data2): ##空间相关系数
    '''data1和data2为2维数组,数据维度相同'''
    data1 = np.array(data1)
    data2 = np.array(data2)
    x,y  = data1.shape
    data1.reshape(1,x*y)
    data2.reshape(1,x*y)
    r  = np.corrcoef(data1,data2)[0,1]
    return r
不知道空间相关系数改成这个对不对？

dongyi

谢谢分享，看看

洗雨

dongyi 发表于 2021-1-15 08:37
谢谢分享，看看

这个不是完整版，没有做检验，谨慎使用

zihan6688

发个带有示例数据的完整版呗

nuist2017njh

现在也在做这方面的内容，学习

西红柿毛毛雨

  r  = np.corrcoef(data1[:,i],data2[:,i])[0,1]

请问都后面的[0,1]是什么意思？

西红柿毛毛雨

shower107 发表于 2021-10-4 17:20
def spacc_corr(data1,data2): ##空间相关系数
    '''data1和data2为2维数组,数据维度相同'''
    data1 ...

  r  = np.corrcoef(data1[:,i],data2[:,i])[0,1]

请问都后面的[0,1]是什么意思？