how to create numpy array using linespace function