2012-06-01 83 views
2

函數func計算大小爲n(即移動窗口寬度爲2的平均數)的數組的後續元素的平均大小的函數是什麼func相鄰數組的平均數

func(numpy.array([1,2,3,4,5])) 
# return numpy.array([1.5, 2.5, 3.5, 4.5]) 

回答

4

無需此功能:

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) 
2
>>> x = np.array([1,2,3,4,5]) 
>>> np.vstack([x[1:], x[:-1]]).mean(axis=0)