[開發技巧]·Python極簡實現滑動平均濾波（基於Numpy.convolve）

​

1.滑動平均概念

滑動平均濾波法（又稱遞推平均濾波法），時把連續取N個取樣值看成一個佇列 ，佇列的長度固定為N ，每次取樣到一個新資料放入隊尾,並扔掉原來隊首的一次資料.(先進先出原則)  把佇列中的N個數據進行算術平均運算,就可獲得新的濾波結果。N值的選取：流量，N=12；壓力：N=4；液麵，N=4~12；溫度，N=1~4 

優點：  對週期性干擾有良好的抑制作用，平滑度高  適用於高頻振盪的系統  

缺點：  靈敏度低  對偶然出現的脈衝性干擾的抑制作用較差  不易消除由於脈衝干擾所引起的取樣值偏差  不適用於脈衝干擾比較嚴重的場合  比較浪費RAM  

2.解決思路

可以發現滑動平均濾波法計算很類似與一維卷積的工作原理，滑動平均的N就對應一維卷積核大小（長度）。

步長會有些區別，滑動平均濾波法滑動步長為1，而一維卷積步長可以自定義。還有區別就是一維卷積的核引數是需要更新迭代的，而滑動平均濾波法核引數都是一。

我們應該怎麼利用這個相似性呢？其實也很簡單，只需要把一維卷積核大小（長度）和N相等，步長設定為1，核引數都初始為1就可以了。由於一維卷積具備速度快，然後我們就可以使用一維卷積來實現這個功能了，快速高效。

使用深度學習框架實現這個功能是否有些大材小用了？是有些大材小用了，因為這裡使用卷積的核引數不用更新，其實沒必要使用複雜的深度學習框架，如果Numpy中可以實現這些功能就更簡單方便了。

說幹就幹，經過查詢發現Numpy.convolve可以實現我們想要的功能。

3.Numpy.convolve介紹

numpy.convolve(a, v, mode=‘full’)
引數：
　　　　a:(N,)輸入的一維陣列
　　　　v:(M,)輸入的第二個一維陣列
　　　　mode:{‘full’, ‘valid’, ‘same’}引數可選

和一維卷積引數類似，a就是被卷積資料，v是卷積核大小。

4.演算法實現

def np_move_avg(a,n,mode="same"):
    return(np.convolve(a, np.ones((n,))/n, mode=mode))

原理說明

執行平均值是卷積數學運算的一個例子。對於執行平均值，沿著輸入滑動視窗並計算視窗內容的平均值。對於離散的1D訊號，卷積是相同的，除了代替計算任意線性組合的平均值，即將每個元素乘以相應的係數並將結果相加。那些係數，一個用於視窗中的每個位置，有時稱為卷積核。現在，N值的算術平均值是(x_1 + x_2 + ... + x_N) / N，所以相應的核心是(1/N, 1/N, ..., 1/N)，這正是我們通過使用得到的np.ones((N,))/N。

邊緣處理

該mode的引數np.convolve指定如何處理邊緣。在這裡選擇了same模式，這樣可以保證輸出長度一種，但你可能還有其他優先事項。這是一個說明模式之間差異的圖：

import numpy as np
import matplotlib.pyplot as plt

def np_move_avg(a,n,mode="same"):
    return(np.convolve(a, np.ones((n,))/n, mode=mode))

modes = ['full', 'same', 'valid']
for m in modes:
    plt.plot(np_move_avg(np.ones((200,)), 50, mode=m));

plt.axis([-10, 251, -.1, 1.1]);

plt.legend(modes, loc='lower center');

plt.show()

​

5.參考

1.https://stackoverflow.com/questions/13728392/moving-average-or-running-