arange函数用于创建等差数组(这里的数组指ndarray数组)。
np.arange([start, ]stop, [step])
# start:起点
# stop:终点
# step:步长
# 一个参数时,参数值为终点,起点取默认值0,步长取默认值1
# 两个参数时,第一个参数为起点,第二个参数为终点,步长取默认值1
# 三个参数时,第一个参数为起点,第二个参数为终点,第三个参数为步长,步长可以是小数
>>> import numpy as np
>>> np.arange(7)
array([0, 1, 2, 3, 4, 5, 6])
>>> np.arange(1, 7)
array([1, 2, 3, 4, 5, 6])
>>> np.arange(0, 2, 0.5)
array([0. , 0.5, 1. , 1.5])