在Python的NumPy库中,arange
函数用于生成一个等差数列。它的定义如下:
numpy.arange([start,] stop[, step,], dtype=None)
参数说明:
start
(可选):等差数列的起始值,默认值为0。stop
(必需):等差数列的结束值(不包括该值),必须指定。step
(可选):等差数列中相邻两项之间的差,默认值为1。如果指定了step
,则start
值会自动调整为start + step
。dtype
(可选):返回值的类型,默认值为float64
。可以指定为其他数值类型,如int8
、int16
、float32
等。示例:
import numpy as np
# 生成一个从0到9的等差数列,步长为2
arr = np.arange(0, 10, 2)
print(arr) # 输出:[0 2 4 6 8]
# 生成一个从2到18的等差数列,步长为3
arr2 = np.arange(2, 19, 3)
print(arr2) # 输出:[ 2 5 8 11 14 17]