【发布时间】:2019-03-30 02:21:59
【问题描述】:
我试图通过对两组数据进行傅立叶变换并将第一个 fft 与第二个 fft 的共轭相乘,然后再转换回时空,来交叉关联两组数据。为了测试我的代码,我将输出与 numpy.correlate 的输出进行比较。但是,当我绘制我的代码时(仅限于某个窗口),似乎这两个信号的方向相反/镜像为零。
这就是我的输出的样子
我的代码:
import numpy as np
import pyplot as plt
phl_data = np.sin(np.arange(0, 10, 0.1))
mlac_data = np.cos(np.arange(0, 10, 0.1))
N = phl_data.size
zeroes = np.zeros(N-1)
phl_data = np.append(phl_data, zeroes)
mlac_data = np.append(mlac_data, zeroes)
# cross-correlate x = phl_data, y = mlac_data:
# take FFTs:
phl_fft = np.fft.fft(phl_data)
mlac_fft = np.fft.fft(mlac_data)
# fft of cross-correlation
Cw = np.conj(phl_fft)*mlac_fft
#Cw = np.fft.fftshift(Cw)
# transform back to time space:
Cxy = np.fft.fftshift(np.fft.ifft(Cw))
times = np.append(np.arange(-N+1, 0, dt),np.arange(0, N, dt))
plt.plot(times, Cxy)
plt.xlim(-250, 250)
# test against convolving:
c = np.correlate(phl_data, mlac_data, mode='same')
plt.plot(times, c)
plt.show()
(两个数据集都用 N-1 个零填充)
【问题讨论】:
-
如果给出了一个最小的可重现示例,您可能会收到回复。例如,数据是什么样的?
-
我已将数据更改为简单的 sin/cos 并将其添加到我的代码中,并带有更新的图像
标签: python numpy fft cross-correlation