配列関数の作成
移動平均の計算¶
def moving_average(a,n=3):
ret = np.cumsum(a,dtype=float)
ret[n:] = ret[n:] - ret[:-n]
return ret[n - 1:]/ n
a = np.arange(10)
print(a)
print(moving_average(a,4))
結果
[0 1 2 3 4 5 6 7 8 9]
[ 1.5 2.5 3.5 4.5 5.5 6.5 7.5]