Python中的 numpy.arange()

简介

numpy 是 Python 中一个重要的科学计算工具库，其中的 arange() 函数可以创建等差数列。

用法

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

• start (可选参数)：起始值，默认为 0。
• stop：终止值，不含该值。
• step (可选参数)：步长，默认为 1。
• dtype (可选参数)：返回结果的数据类型，默认根据给定参数自动匹配。

返回结果

返回值是一个 numpy.ndarray 类型的一维数组。

import numpy as np

# 默认情况下，从 0 开始，步长为 1，生成长度为 10 的等差数列
a = np.arange(10)
print(a)  # [0 1 2 3 4 5 6 7 8 9]

# 可以指定起始值和步长
b = np.arange(2, 10, 2)
print(b)  # [2 4 6 8]

# 给出起始值和终止值（终止值不包括在范围内），不指定步长时默认为 1
c = np.arange(3, 10)
print(c)  # [3 4 5 6 7 8 9]

# 指定返回值的数据类型
d = np.arange(5, dtype=float)
print(d.dtype)  # float64

注意事项

• 参数类型不是必须为整数，可以是浮点型。
• 起始值和终止值可以是负数。
• 步长可以是负数，此时从大到小生成等差数列。
• start 和 stop 参数不一定都要传递，但要保证传入的参数个数正确。