Подтвердить что ты не робот

Разница между соседними элементами

У меня есть алгоритм вычисления разности между соседними элементами в чистом питоне:

    a = range(1000000) #it numpy array in my case
    prev = a[0]
    b = [0, ]
    for i in a[1:]:
        b.append(i - prev)
        prev = i

Можно ли переписать эти функции с помощью Numpy?

4b9b3361

Ответ 1

Существует метод diff:

a = range(5) # python list of numpy array
np.diff(a)

возвращает

array([1, 1, 1, 1])