在 Numpy 中计算单位阶跃函数


要计算单位阶跃函数,在 Python Numpy 中使用 numpy.heaviside() 方法。第 1 个参数是输入数组。第 2 个参数是数组元素为 0 时的函数值。返回输出数组,element-wise Heaviside step function of x1。如果 x1 和 x2 都是标量,则这是一个标量。

单位阶跃函数定义为 −

0 if x1 < 0
heaviside(x1, x2) = x2 if x1 == 0
1 if x1 > 0

其中 x2 通常取为 0.5,但也经常使用 0 和 1。

步骤

首先,导入必需的库 −

import numpy as np

使用 array() 方法用浮点类型创建一个数组 −

arr = np.array([50.8, -3.5, 120.3, 0, 320.1, 450.4, 0])

显示数组 −

print("Array...
", arr)

获取数组的类型 −

print("
Our Array type...
", arr.dtype)

获取数组的维度 −

print("
Our Array Dimension...
",arr.ndim)

获取数组的形状 −

print("
Our Array Shape...
",arr.shape)

要计算单位阶跃函数,在 Python Numpy 中使用 numpy.heaviside() 方法。第 1 个参数是输入数组。第 2 个参数是数组元素为 0 时的函数值 −

print("
Result...
",np.heaviside(arr, 1))

示例

import numpy as np

# Create an array with float type using the array() method
arr = np.array([50.8, -3.5, 120.3, 0, 320.1, 450.4, 0])

# Display the array
print("Array...
", arr) # Get the type of the array print("
Our Array type...
", arr.dtype) # Get the dimensions of the Array print("
Our Array Dimension...
",arr.ndim) # Get the shape of the Array print("
Our Array Shape...
",arr.shape) # To compute the Heaviside step function, use the numpy.heaviside() method in Python Numpy # The 1st parameter is the input array # The 2nd parameter is the The value of the function when array element is 0 print("
Result...
",np.heaviside(arr, 1))

输出

Array...
[ 50.8 -3.5 120.3 0. 320.1 450.4 0. ]

Our Array type...
float64

Our Array Dimension...
1

Our Array Shape...
(7,)

Result...
[1. 0. 1. 1. 1. 1. 1.]

更新于:2022-02-08

668 次浏览

开启你的 职业

完成课程并获得认证

开始
广告