numpy 单位阶跃函数

numpy中没有直接实现单位阶跃函数的函数，但可以通过以下方式实现：

import numpy as np

def unit_step(x):
    return np.piecewise(x, [x < 0, x >= 0], [0, 1])

这里使用了numpy的piecewise函数，根据输入x的值，返回不同的值。当x小于0时，返回0；当x大于等于0时，返回1。这样就实现了单位阶跃函数。