python 平滑函数 smooth

## 2021年10月18日13:14:28 完成
def smooth2nd(x,M): ##x 为一维数组
    K = round(M/2-0.1) ##M应为奇数，如果是偶数，则取大1的奇数
    lenX = len(x)
    if lenX<2*K+1:
        print('数据长度小于平滑点数')
    else:
        y = np.zeros(lenX)
        for NN in range(0,lenX,1):
            startInd = max([0,NN-K])
            endInd = min(NN+K+1,lenX)
            y[NN] = np.mean(x[startInd:endInd])
##    y[0]=x[0]       #首部保持一致
##    y[-1]=x[-1]     #尾部也保持一致
    return(y)
 
 
######################################################################
## 测试程序
if __name__ == '__main__':
    import numpy as np
    import matplotlib.pyplot as plt
    mngr = plt.get_current_fig_manager()
##    mngr.window.wm_geometry("+70+0")  # 调整窗口在屏幕上弹出的位置
     
    x = np.random.randn(100,1)  ##产生随机数
    M=5 ## 做 M点平滑, M为奇数
    y=smooth2nd(x,M)
     
    ## 画图展示
    plt.plot(x,c='b')
    plt.plot(y,c='r')
    plt.show()