「PythonとJavaScriptではじめるデータビジュアライゼーション」を読む

配列関数の作成

移動平均の計算¶
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]