我在哪里引用这个类型,numpy.array(x> 0,dtype = np.int),函数?

时间:2017-06-30 11:11:26

标签: arrays numpy

当我使用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()

0 个答案:

没有答案