У меня есть массив с большими числами, который мне нужно манипулировать так, чтобы каждый элемент был изменен либо на 1, либо на 0, если условие выполнено (будет использоваться в качестве пиксельной маски позже). В массиве насчитывается около 8 миллионов элементов, и мой текущий метод слишком долго подходит для конвейера восстановления:
for (y,x), value in numpy.ndenumerate(mask_data):
if mask_data[y,x]<3: #Good Pixel
mask_data[y,x]=1
elif mask_data[y,x]>3: #Bad Pixel
mask_data[y,x]=0
Есть ли функция numpy, которая ускорит это?