numpy中的任何剪切功能？

Question

y=np.array([0.4,0.5,0.6])
threshold=0.5

print(list(map(lambda x:1 if x>threshold else 0, y)))
[0, 0, 1]

numpy中有任何功能可以做这样的工作吗？

Answer 1

numpy.where()怎么样，按照以下几行：

numpy.where(y>threshold, 1, 0)

产生

array([0, 0, 1])