使用Numpy快速去除信号趋势项

1、引言

在我们采集传感器信号的时候，由于传感器受温度影响，使得采集到的信号存在频移现象(趋势项)，这种频移现象往往影响对信号的分析和识别。所以我们要消除这种趋势项。

2、采用多项式拟合的方法

原理暂时pass

3、使用Numpy去除信号趋势项

import numpy as np
import matplotlib.pyplot as plt

def trend_remove(x, y):  # 
    param = np.polyfit(x, y, 2)   #用2次多项式拟合x，y数组,返回多项式系数
    y_poly_func = np.poly1d(param)#拟合完之后，用生成的多项式系数用来生成多项式函数
    y_poly = y_poly_func(x) #生成多项式函数之后，就是获取x在这个多项式处的值
    detrend_y = y - y_poly  # 原始信号减去拟合函数的值即为去除趋势项后的信号
    
    plt.figure(figsize=(20,8))
    plt.subplot(131)
    plt.title('原始信号')
    plt.plot(x, y)
    plt.subplot(132)
    plt.title('多项式拟合')
    plt.plot(x, y)
    plt.plot(x, y_poly)
    plt.subplot(133)
    plt.title('去除趋势项后信号')
    plt.plot(x, detrend_y)
    plt.show()