2012-06-01 1 views
2

Что такое функция func, которая вычисляет n-1 -размерную матрицу средних значений последовательных элементов массива размера n (т. Е. Скользящее среднее с шириной окна 2)?массив средних значений соседей

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) 

 Смежные вопросы

  • Нет связанных вопросов^_^