Средние вычисления для группы соседних точек в строке

Question

Я пытаюсь вычислить среднее значение для группы соседних точек в массиве, с которым я могу легко перемещаться. Я советовал определить вектор называется «трястись» целые числа, например, от 5 до 15. Heres кода я прямо сейчас:

import matplotlib.pyplot as plt 
import numpy as np 
from scipy.optimize import curve_fit 

x = np.arange(0,10,.05) 
wave = np.sin(x) 

noise = np.random.normal(0,.05,200) 
y = noise + wave 
noise2 = np.random.normal(0,.05,200) 
y2 = noise2 + wave 
noise3 = np.random.normal(0,.05,200) 
y3 = noise3 + wave 


y_avg = (np.array(y) + np.array(y2) + np.array(y3))/3 

for i in range(5, 15): 
    mean = np.mean(y_avg) 
print mean 


plt.plot(x, y, color = 'red', label= 'line 1') 
plt.plot(x, y2, color = 'pink', label = 'line 2') 
plt.plot(x, y3, color = 'magenta', label = 'line 3') 
plt.plot(x, y_avg, color = 'blue', label = 'average') 
plt.legend() 
plt.show() 

Есть ли лучший способ сделать это, что предполагает индексацию вектор данных/как это сделать?

Answer 1

Использование нарезка вектора:

mean = np.mean(y_avg[5:15]) 

Если вы хотите среднюю точку, это работает даже размер окна:

def mean_slice(midp, size, array): 
    slice_index = [midp - (size/2), midp + (size/2)] 
    # Remove out of bounds indexes 
    slice_index = [i if i > 0 else 0 for i in slice_index] 
    slice_index = [i if i < len(array) else 0 for i in slice_index] 
    return np.mean(array[slice(*slice_index)]) 

print mean_slice(1, 4, a) 
>>>1.0