函数func
是什么函数计算n-1
- 大小为n
的数组的后续元素的平均大小数组(即窗口宽度为2的移动平均值)?
func(numpy.array([1,2,3,4,5]))
# return numpy.array([1.5, 2.5, 3.5, 4.5])
答案 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)