“В массиве NP, как сделать элемент как 1, если он превышает порог” Ответ

В массиве NP, как сделать элемент как 1, если он превышает порог

a = [0 if a_ > thresh else a_ for a_ in a]