Change to ```import numpy as np```, include ```import numpy.ma as ma``` and deal with fall-out.