尝试用函数计算自相关系数

Trying to calculate autocorrelation coefficient with function

我正在尝试复制这个自相关公式,其中 T 是样本数,k 是滞后数。

根据定义,第一个元素的自相关为 1.0。对于其余元素,我们使用列表理解(请注意,问题中的公式假定索引从 1 到 T,而 python 数组是基于 0 的):

mean = np.mean(x)    
denominator = np.sum((x-mean)**2)
[1.0] + [np.sum((x[j:]-mean) * (x[:-j]-mean)) / denominator for j in range(1, len(x))]