python linspace function to create arrays