错误而从功能

Question

生成列表我有这样

def auto_correlation(dat, k): 
    s = np.mean(dat) 
    v = np.var(dat) 
    N = dat.size 
    sum = np.dot((dat[:N - k] - s), (dat[k:] - s)) 
    return 1/v/(N - k) * sum 

的功能，并希望从它创建一个列表这样

autocorrOfTimeSeries = [auto_correlation(time_series,k) for k in range(50001)] 

我总是得到错误

ValueError: shapes (4999,) and (0,) not aligned: 4999 (dim 0) != 0 (dim 0) 

虽然。时间序列仅仅是一个1 d阵列，以及auto_correlation返回单个值。

Answer 1

的错误可能是因为在某些时候切片[:N-k]和[k:]返回不同的形状。

的一个普遍问题是，使用矢量（NumPy的基于阵列）操作，则可能需要垫0您的数据系列。您还可以看看NumPy相关函数https://docs.scipy.org/doc/numpy/reference/generated/numpy.correlate.html#numpy.correlate或SciPy的基于FFT的相关函数https://docs.scipy.org/doc/scipy/reference/tutorial/signal.html#convolution-correlation