Numpy Apply Function To 1d Array