Eu tenho uma grande matriz numpy que preciso manipular para que cada elemento seja alterado para 1 ou 0 se uma condição for atendida (será usada como uma máscara de pixel posteriormente). Existem cerca de 8 milhões de elementos na matriz e meu método atual leva muito tempo para o pipeline de redução:
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
Existe uma função numpy que aceleraria isso?
mask_data[y,x]==3
?