Python Select Part Of Numpy Linspace