numpy.arange()
函数用于生成等差数列数组。它的基本语法如下:
numpy.arange([start,] stop[, step,], dtype=None)
参数说明:
start
(可选):等差数列的起始值,默认值为 0。stop
(必需):等差数列的结束值(不包括该值),必须指定。step
(可选):等差数列的公差,默认值为 1。如果未指定,则生成连续整数序列。dtype
(可选):生成的数组的元素类型,默认值为 float64
。示例:
import numpy as np
# 生成一个从 0 到 9 的等差数列数组,公差为 2
arr1 = np.arange(0, 10, 2)
print(arr1) # 输出:[0 2 4 6 8]
# 生成一个从 5 到 -5 的等差数列数组,公差为 -1
arr2 = np.arange(5, -5, -1)
print(arr2) # 输出:[ 5 4 3 2 1 0 -1 -2 -3 -4]
# 生成一个包含 1 到 10 的整数数组,元素类型为 int32
arr3 = np.arange(1, 11, dtype=np.int32)
print(arr3) # 输出:[1 2 3 4 5 6 7 8 9 10]