サイズの配列の結果の要素の平均のサイズの配列func
を計算する関数は何ですか(つまり、ウィンドウ幅2の移動平均)?n-1
n
func(numpy.array([1,2,3,4,5]))
# return numpy.array([1.5, 2.5, 3.5, 4.5])
ここでは関数は必要ありません:
import numpy as np
x = np.array([1,2,3,4,5])
x_f2 = 0.5*(x[1:] + x[:-1])
関数として必要な場合:
def window(x, n):
return (x[(n-1):] + x[:-(n-1)])/float(n)
>>> x = np.array([1,2,3,4,5])
>>> np.vstack([x[1:], x[:-1]]).mean(axis=0)