среднее значение элементов i и i + 1 в массиве numpy

Question

Из любопытства, есть ли конкретная функция numpy для выполнения следующих действий (которые предположительно будут быстрее):

a = np.array((0,2,4))
b = np.zeros(len(a) - 1)
for i in range(len(b)):
    b[i] = a[i:i+2].mean()

print(b)
#prints [1,3]

Приветствия

Answer 1

Вы можете использовать

b = (a[1:] + a[:-1]) / 2.

, чтобы избежать цикла Python.