Numpy Array Operation On Function