numpy使用阶跃函数

NumPy中没有内置的阶跃函数，但可以使用numpy.heaviside()函数来实现。numpy.heaviside()函数的定义如下：

numpy.heaviside(x1, x2, out=None)

其中，x1表示输入的数组，x2表示阈值，out表示输出的数组。如果x1[i]>=x2，则out[i]=1；如果x1[i]<x2，则out[i]=0；当x1[i]=0时，out[i]=0.5。

下面是一个使用numpy.heaviside()函数实现阶跃函数的例子：

import numpy as np

def step_function(x):
    return np.heaviside(x, 0)

print(step_function(np.array([-1.0, 1.0, 2.0])))

输出结果为：

[0. 1. 1.]

这个例子中，x传入的是一个包含三个元素的数组，输出结果为0，1，1，分别对应x=-1.0，x=1.0，x=2.0。