Untitled
unknown
plain_text
3 years ago
385 B
7
Indexable
def corrfft(x): N = len(x) x = np.pad(x, ((0, N))) fft = np.fft.fft(x) iff = np.fft.ifft(fft*np.conjugate(fft)) iff = np.concatenate([iff[N+1:], iff[:N]]) return iff.real import numpy as np from scipy import signal t = np.linspace(0, 10, 100) s = np.cos(t) ac = signal.correlate(s,s) ac_by_fft = corrfft(s) np.allclose(ac, ac_by_fft) Out[99]: True
Editor is loading...