matlab里有一个smooth函数,有滑动平均的功能,用起来很丝滑。但python里面似乎没有找到自带的类似函数,自己写了一个。希望尽可能少的调用外部模块。
请大家帮忙看看,更欢迎拿过去用一用,看哪样修改会更好一点,效率高一点。
## 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)
######################################################################
## 测试程序
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()



