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

#python #arrays #numpy

#python #массивы #numpy

Вопрос:

Из любопытства, существует ли конкретная функция 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]
  

Приветствия

Ответ №1:

Вы могли бы использовать

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

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

Комментарии:

1. Аккуратно, мне это нравится 🙂 Думаю, я не слишком долго работал с numpy! Спасибо!