格兰杰因果计算python

#使用格兰杰因果关系
def cal_grangercausality(array):
    arr=np.zeros((array.shape[0],array.shape[0]))
    for i in range(array.shape[0]):
       for j in range(array.shape[0]):
           #构建两列数据
           tes=np.vstack((array[i], array[j]))
           tes=tes.T
           rels=grangercausalitytests(tes, maxlag=1, addconst=True, verbose=True)
           d=rels[1][0]['params_ftest'][1]
           arr[i][j]=d
    return arr