当我使用numpy.array(x > 0, dtype=np.int)
时,如果满足1
或者返回x > 0
,我想函数会返回0
。我在哪里可以参考这个功能的文档?我试图在numpy文档中找到这个函数(https://docs.scipy.org/doc/numpy/reference/generated/numpy.array.html),但我找不到它。
import numpy as np
import matplotlib.pylab as plt
def step_function(x):
return np.array(x > 0, dtype=np.int)
x = np.arange(-5.0, 5.0, 0.001)
y = step_function(x)
plt.plot(x, y)
plt.ylim(-0.1, 1.1)
plt.show()