Step関数について、ふたつ書いてみた。
ひとつは、plt.plotの引数に、ループした値を代入させている。
もうひとつは、Step関数内を工夫している。x>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()
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()