python - 非周期函数与 NumPy 的互相关

Question

我有两个数据集，我正试图将它们相互关联。它们看起来类似于 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 坐标，应该在两个信号最大重叠的地方找到。

关于如何为这种函数计算这个数量有什么想法吗？

编辑:我应该补充一点，对于我的真实数据，所有值都将在零和一之间

编辑编辑:以下函数实际上更像我的真实数据:

x = linspace(-15, 15, 400)
f1 = (arctan(-x) + pi/2) / pi
f2 = (arctan(-x + 2) + pi/2) / pi

因此使用此处给出的公式:http://paulbourke.net/miscellaneous/correlate/我可以编写一个互相关函数来填充数据以在左侧添加 1，在右侧添加 0:

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 中编写我的算法？

Answer 1

正如人们所指出的，互相关被数据之外的填充所混淆。

虽然感觉就像您丢弃了好的数据，但通常最好只修剪数据集，这样就可以在没有假设的情况下完成关联(至少与将实际数据与虚构数据相关联的替代方案相比)填充)。

x = linspace(-15, 15, 4000)
f1 = (arctan(-x) + pi/2) / pi
f2 = (arctan(-x + 2) + pi/2) / pi

L4 = int(len(f2)/8)
sf2 = f2[L4:-L4]

c = correlate(f1-mean(f1), sf2-mean(f1), 'same')
print "peak correlation occurs at:", x[argmax(c)]  # -2.02925731433

plt.plot(x, c)
plt.show()

此外，我不确定 xcorr 是否是这里的最佳方法。将不同偏移的 y 轴值之间的距离相加并取最小值如何，这样可以避免零位置等所有问题。