time_analysis.py 文件源码

def corr(data):
  ns = data.shape[0];
  nt = data.shape[1];

  mean = np.nanmean(data, axis = 0);
  std = np.nanstd(data - mean, axis = 0);

  c = np.zeros((nt, nt));
  for t1 in range(nt):
    #for t2 in range(nt):
      #c[t1,t2] = np.nanmean( (data[:,t1] - mean[t1]) * (data[:,t2] - mean[t2]), axis = 0); # / (std[t1] * std[t2]);
    c[t1,:] = np.nanmean( (data[:,:].T * data[:, t1]).T, axis = 0);
  return c;