【发布时间】:2017-08-11 15:02:31
【问题描述】:
我有一个非常大的形状 (T, N) 数据集,我想在 Python 中计算自相关函数,例如 numpy.correlate:
c[k] = sum_n sum_t a[t, n] * a[t+k, n]
但使用 for 循环对所有样本 N 和不带 (!) 求和。根据“有效”模式进行计算就足够了。但是,此函数仅允许一维数组。有没有快速的方法来做到这一点?
【问题讨论】:
-
不幸的是,该函数计算所有互相关,也包括不同样本之间的相关。但是,我想只在 T 方向计算相关性,同时在 N 方向求和。
-
correlate1d 和每行应用怎么样:docs.scipy.org/doc/scipy-0.16.0/reference/generated/…?
-
嗯,可能,但仍然是一个 for 循环:/
标签: python numpy vectorization correlation