在 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.]
广告