Numpy Array From Function