Step関数について、ふたつ書いてみた。
ひとつは、plt.plotの引数に、ループした値を代入させている。
もうひとつは、Step関数内を工夫している。x>0は、0より大きい、0以下で、True/Falthとなっているので、astypeメソッドを使って、bool型をinit型に変換させている。
import numpy as np
import matplotlib.pyplot as plt
x=np.arange(-10,11)
def step_function(a):
if a >0:
return 1
else:
return 0
plt.plot(x,[step_function(i) for i in x])
plt.show()
![](https://tsurumaki-office.com/wp-content/uploads/2021/03/スクリーンショット-2021-03-15-11.52.14.png)
import numpy as np
import matplotlib.pyplot as plt
x=np.arange(-10,11)
def step_function(a):
return (x>0).astype(np.int8)
plt.plot(x,step_function(x))
plt.show()
![](https://tsurumaki-office.com/wp-content/uploads/2021/03/スクリーンショット-2021-03-15-11.52.14.png)