Функция diff
Функция diff вычисляет разности между соседними элементами массива.
По умолчанию вычисляется разность первого порядка (соседние элементы вычитаются).
Первым параметром функция принимает массив, вторым - порядок разности (по умолчанию 1),
третьим - ось (по умолчанию последняя ось).
Результат содержит на один элемент меньше вдоль оси, по которой вычисляется разность.
Синтаксис
np.diff(a, n=1, axis=-1)
Пример
Вычислим разность первого порядка для одномерного массива:
import numpy as np
arr = np.array([1, 3, 6, 10, 15])
res = np.diff(arr)
print(res)
Результат выполнения кода:
[2 3 4 5]
Пример
Вычислим разность второго порядка для того же массива:
import numpy as np
arr = np.array([1, 3, 6, 10, 15])
res = np.diff(arr, n=2)
print(res)
Результат выполнения кода:
[1 1 1]
Пример
Вычислим разность вдоль столбцов (ось 0) для двумерного массива:
import numpy as np
arr = np.array([[1, 2, 3],
[4, 6, 8],
[9, 12, 15]])
res = np.diff(arr, axis=0)
print(res)
Результат выполнения кода:
[[3 4 5]
[5 6 7]]
Пример
Вычислим разность вдоль строк (ось 1) для двумерного массива:
import numpy as np
arr = np.array([[1, 2, 3],
[4, 6, 8],
[9, 12, 15]])
res = np.diff(arr, axis=1)
print(res)
Результат выполнения кода:
[[1 1]
[2 2]
[3 3]]
