In [ ]:
import numpy as np
import matplotlib.pylab as plt

def step_function(x):
    return np.array(x > 0, dtype=np.int) # 与えられた配列に対して比較演算を行い、それをbooleanに変換している

x = np.arange(-5.0, 5.0, 0.1)
y = step_function(x)
plt.plot(x, y)
plt.ylim(-0.1, 1.1) #yの範囲を指定
plt.show()

In [ ]: