Numpy Mask An Array In Math