我有两个数据集要相互关联。它们看起来与arctan
函数相似,因此我一直将其用作模型来研究如何进行信号处理。
x = linspace(-15, 15, 2**13)
f1 = arctan(x)
f2 = arctan(x + 2)
我需要回答的问题是,我需要将绿色信号移动多少才能使其(大部分)与蓝色信号重叠?f1
我认为这就像在和的互相关函数中找到最大值一样简单f2
,我在这里广泛遵循了建议:如何将两个时间序列与间隙和不同的时间基准相关联?. 这是我一直在尝试的
c = correlate(f1, f2, 'full')
s = arange(1-2**13, 2**13)
dx = 30/2**13
shift = s[c.argmax()]*dx
我希望shift
或多或少正好等于 2,但实际上它只是0.234
. 这对我来说没有任何意义;我找到了互相关最大值的 x 坐标,应该在两个信号最大重叠的地方找到。
关于如何为这种函数计算这个数量的任何想法?
编辑:我应该添加,对于我的真实数据,所有的值都将在零和一之间
编辑编辑:以下函数实际上更像我的真实数据:
x = linspace(-15, 15, 400)
f1 = (arctan(-x) + pi/2) / pi
f2 = (arctan(-x + 2) + pi/2) / pi
因此,使用此处给出的公式:http: //paulbourke.net/miscellaneous/correlate/我可以编写一个互相关函数来填充数据以在左侧添加 1 并在右侧添加零:
def xcorr(x, y);
mx = x.mean()
my = y.mean()
sx = x.std()
sy = y.std()
r = zeros(2*len(x))
for d in range(-len(x), len(x)):
csum = 0
for i in range(0, len(x):
yindx = i - d
if i - d < 0:
yval = 1
elif i - d >= len(x):
yval = 0
else:
yval = y[yindx]
csum += (x[i] - mx) * (yval - my)
r[d + len(x)] = csum / (sx * sy)
return r
有了这个功能,我现在可以做
c = xcorr(f1, f2)
s = arange(-400, 400)
dx = 30/400
shift = s[c.argmax()] * dx
结果是 2.025,以这种精度可以接近 2。所以看起来 Jamie 是正确的,问题在于 numpy 如何correlate
填充信号。
所以,显然我的xcorr
功能真的很慢。现在的问题是,有没有办法让 NumPy 做类似的事情,或者我应该继续使用 C 语言编写我的算法ctypes
?