NumPy 函数

NumPy arange() 函数返回给定间隔内均匀间隔的值。这些值在 [start, stop) 范围内以指定的步长生成。

语法

numpy.arange(start, stop, step, dtype=None)

参数

start可选。 指定间隔的开始（含）。默认值为 0。
stop必填。 指定间隔结束（不包括）。
step可选。 指定步长。默认值为 1。
dtype可选。 指定输出数组的类型。如果未给出 dtype，则从其他输入参数推断数据类型。

返回值

返回均匀间隔的数组。

示例:创建数组

在下面的示例中，该函数用于使用 arange() 函数创建一维数组。

import numpy as np

#使用范围[10,55)创建数组并且步长=5
Arr1 = np.arange(10,55,5)
print("Arr1 is:", Arr1)

#使用范围[10,16)创建数组并且步长=1且dtype=float
Arr2 = np.arange(10,16,dtype=float)
print("Arr2 is:", Arr2)

上述代码的输出将是:

Arr1 is: [10 15 20 25 30 35 40 45 50]
Arr2 is: [10. 11. 12. 13. 14. 15.]

示例:创建矩阵

使用 reshape 函数，可以将数组转换成矩阵。考虑以下示例。

import numpy as np

#使用范围[10,55)创建数组并且步长=5
Arr = np.arange(10,55,5)
print("Arr is:", Arr)

#重塑 Arr
Arr = Arr.reshape(3,3)
print("\nArr is:")
print(Arr)

上述代码的输出将是:

Arr is: [10 15 20 25 30 35 40 45 50]

Arr is:
[[10 15 20]
 [25 30 35]
 [40 45 50]]