【发布时间】:2013-04-06 01:04:26
【问题描述】:
我有两个数据集要相互关联。它们看起来类似于 arctan 函数,所以我一直将它用作模型来研究如何进行信号处理。
x = linspace(-15, 15, 2**13)
f1 = arctan(x)
f2 = arctan(x + 2)
我需要回答的问题是,我需要将绿色信号移动多少才能使其(大部分)与蓝色信号重叠?我认为这就像在f1 和f2 的互相关函数中找到最大值一样简单,我在这里大致遵循了以下建议:How to correlate two time series with gaps and different time bases?。这是我一直在尝试的
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 坐标,应该在两个信号最大重叠的地方找到。
关于如何为这种函数计算这个数量的任何想法?
编辑:我应该添加,对于我的真实数据,所有值都将在零和一之间
EDIT EDIT:以下函数其实更像我的真实数据:
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 做类似的事情,还是应该继续使用 ctypes 用 C 语言编写我的算法?
【问题讨论】:
-
有趣 - 如果我将 1.5 添加到 f1 和 f2 以使它们始终为正,则偏移变为零......
-
@Eike 那么你应该让你的评论成为答案。也许添加一个链接以获得进一步的解释。
-
他们甚至在解释段落中详细说明了信号中正负区域的影响:“这是因为当峰(正区域)对齐时,它们对积分。类似地,当波谷(负区域)对齐时,它们也会对积分做出正贡献,因为两个负数的乘积是正数。”
-
@StephaneRolland 是填充破坏了 OP 的努力,而不是对积极性的不存在要求。
-
只是一个迂腐的评论:
dx应该是30./(2**13 - 1)。
标签: python numpy signal-processing