邻居平均数组

时间:2012-06-01 11:03:22

标签: numpy average

函数func是什么函数计算n-1 - 大小为n的数组的后续元素的平均大小数组(即窗口宽度为2的移动平均值)?

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

2 个答案:

答案 0 :(得分: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)

答案 1 :(得分:2)

>>> x = np.array([1,2,3,4,5])
>>> np.vstack([x[1:], x[:-1]]).mean(axis=0)