Замена элементов Numpy, если условие выполнено ⇐ Python

[Anonymous]

У меня есть большой массив numpy, которым мне нужно манипулировать, чтобы каждый элемент менялся на 1 или 0, если условие выполнено (позже будет использоваться в качестве пиксельной маски). В массиве около 8 миллионов элементов, и мой текущий метод занимает слишком много времени для конвейера сокращения:

Код: Выделить всё

for (y,x), value in numpy.ndenumerate(mask_data):

if mask_data[y,x]3: #Bad Pixel
mask_data[y,x]=0

Есть ли функция numpy, которая могла бы ускорить это?