Любая функция сокращения в 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])