Функция heaviside
Функция heaviside вычисляет ступенчатую функцию Хевисайда для каждого элемента массива.
Первым параметром функция принимает массив значений, для которых вычисляется функция.
Вторым параметром передаётся значение, которое будет возвращено для элементов, равных нулю.
Функция возвращает массив той же формы, что и входной.
Синтаксис
np.heaviside(x, h0)
Пример
Давайте вычислим функцию Хевисайда для массива чисел:
import numpy as np
arr = np.array([-3, -1, 0, 2, 5])
res = np.heaviside(arr, 0.5)
print(res)
Результат выполнения кода:
[0. 0. 0.5 1. 1. ]
Пример
Давайте посмотрим, как меняется результат при изменении значения для нуля:
import numpy as np
arr = np.array([-2, 0, 3])
res1 = np.heaviside(arr, 1)
res2 = np.heaviside(arr, 0)
print("h0 = 1:", res1)
print("h0 = 0:", res2)
Результат выполнения кода:
h0 = 1: [0. 1. 1.]
h0 = 0: [0. 0. 1.]
Пример
Функция heaviside работает и с двумерными массивами:
import numpy as np
arr = np.array([[-4, -1, 0], [2, 5, 0]])
res = np.heaviside(arr, 0)
print(res)
Результат выполнения кода:
[[0. 0. 0.]
[1. 1. 0.]]
