Using NumPy's np.arange() Effectively