數(shù)學原理
　　
在數(shù)字信號處理中，相關(guān)(correlation)可以分為互相關(guān)(cross correlation)和自相關(guān)(auto-correlation). 互相關(guān)是兩個數(shù)字序列之間的運算；自相關(guān)是單個數(shù)字序列本身的運算，可以看成是兩個相同數(shù)字序列的互相關(guān)運算．互相關(guān)用來度量一個數(shù)字序列移位后，與另一個數(shù)字序列的相似程度．其數(shù)學公式如下：

其中，f 和 g 為數(shù)字序列，n 為移位的位數(shù)，f* 表示 f 序列值的復數(shù)共軛，即復數(shù)的實部不變，虛部取反．

而卷積（convolution）與互相關(guān)運算相似，定義為將其中一個序列反轉(zhuǎn)并移位后，兩個序列的乘積的積分（求和），其數(shù)學公式如下：

其中，f 和 g 為數(shù)字序列，n 為移位的位數(shù).

在實數(shù)范圍內(nèi)，f 的復數(shù)共軛 f* = f ．此時，通過比較上面兩式可知：序列 f 與將序列 g 反轉(zhuǎn)后的序列的卷積為序列 f 與序列 g 的互相關(guān)

Python 實現(xiàn)

采用兩種方式實現(xiàn)：自定義互相關(guān)函數(shù)和直接調(diào)用 numpy.correlate 或 numpy.convolve．

在 numpy 中, numpy.correlate 函數(shù)實現(xiàn)兩個一維數(shù)組的互相關(guān)操作；numpy.convolve 實現(xiàn)了兩個一維數(shù)組的卷積操作．其中定義了三種模式（‘valid’, ‘same’,‘full’）.

設(shè)兩個序列長度分別為 M 和 N，則

• ‘valid’ 模式：輸出長度為 max(M,N)-min(M,N)+1．只返回兩個序列完全重合部分的點的卷積或相關(guān)運算；
• ‘same’ 模式：輸出長度為兩個序列中的較長者，即 max(M,N)；
• ‘full’ 模式：輸出長度為 M+N-1, 返回所有包含重疊部分的點.

互相關(guān)或卷積，實際上，就是計算兩個序列（一維數(shù)組）在不同移位情況下，兩個序列逐位相乘之后，求和的結(jié)果．不同模式只是返回互相關(guān)或卷積結(jié)果的不同部分．

注：如果在超出數(shù)組的索引范圍，用 0 填充．

下面代碼，采用自定義函數(shù) correlate_func (只適用于實數(shù)值) 實現(xiàn) numpy.correlate 和 numpy.convolve 的三種模式，并進行測試．

#!//usr/bin/env python # -*- coding: utf8 -*- """ 遇到問題沒人解答？小編創(chuàng)建了一個Python學習交流QQ群：579817333 尋找有志同道合的小伙伴，互幫互助,群里還有不錯的視頻學習教程和PDF電子書！ """ from __future__ import print_function import numpy as npdef correlate_func(a, b, mode='valid', conv=True):'''correlation or convolution in 1-d array with real numbers'''if a is None or b is None:return Noneif len(a) > len(b):# Ensure the length of a is no longer than that of b.return correlate_func(b, a, mode)# Convert to np.array typea, b = list(map(np.array, [a, b]))if conv: a = a[::-1] # if convolution is true, reverse the shorter res = []min_len, max_len = len(a), len(b)if mode == 'valid':output_length = max_len - min_len + 1tmp = belif mode == 'same':output_length = max_lentmp = np.hstack((np.zeros(min_len-1), b))elif mode == 'full':output_length = max_len + min_len - 1tmp = np.hstack((np.zeros(min_len-1), b, np.zeros(min_len-1)))else:raise Exception("No such mode {}!".format(mode))# For each point, get the total sum of element-wise multiplicationfor i in range(output_length):val = np.sum(a * tmp[i:min_len+i])res.append(val)return np.array(res, dtype=a.dtype)def test():a = [1, 2, 3]b = [1, 2]names = ['numpy.correlate', 'correlate_func', 'numpy.convolve', 'correlate_func(convolution)']funcs = [np.correlate, correlate_func, np.convolve, lambda *args: correlate_func(*args, conv=True)]for i, (name, func) in enumerate(zip(names, funcs)[:4]):print ('-----' * 30 if i & 0x01 == 0 else '')print ("{} output result: ".format(name))print (' valid mode: ', func(a, b, 'valid'))print (' same mode: ', func(a, b, 'same'))print (' full mode: ', func(a, b, 'full'))if __name__ == '__main__':test()

除此之外，在 matplotlib.pyplot 模塊中，實現(xiàn)了用于可視化的自相關(guān)函數(shù) matplotlib.pyplot.acorr 和互相關(guān)函數(shù) matplotlib.pyplot.xcorr, 官方網(wǎng)址提供的一個示例代碼如下:

import matplotlib.pyplot as plt import numpy as npnp.random.seed(0)x, y = np.random.randn(2, 100) fig = plt.figure() ax1 = fig.add_subplot(211) ax1.xcorr(x, y, usevlines=True, maxlags=50, normed=True, lw=2) ax1.grid(True) ax1.axhline(0, color='black', lw=2)ax2 = fig.add_subplot(212, sharex=ax1) ax2.acorr(x, usevlines=True, normed=True, maxlags=50, lw=2) ax2.grid(True) ax2.axhline(0, color='black', lw=2)plt.show()

總結(jié)

以上是生活随笔為你收集整理的相关与卷积（数字信号处理）的数学原理及 Python 实现的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。