阶跃函数的Python实现 ← 斋藤康毅

【阶跃函数的Python代码】
下面的 step_function 函数，给出了阶跃函数的两种 Python 实现。
一种是 return np.array(x>0,dtype=np.int)，另一种是 return (x>0).astype(np.int)。

import numpy as np
import matplotlib.pylab as plt

def step_function(x):
    return np.array(x>0,dtype=np.int)
    # return (x>0).astype(np.int)

x=np.arange(-5.0,5.0,0.1)
y=step_function(x)

plt.plot(x,y)
plt.ylim(-0.1,1.1)
plt.show()

【阶跃函数的图形】