python numpy.arange如何使用

numpy.arange()是NumPy库中的一个函数，用于生成等差数列。它的基本语法如下：

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

参数说明：

• start（可选）：序列的起始值，默认值为0。
• stop（必需）：序列的结束值（不包括该值），必须指定。
• step（可选）：序列中相邻元素之间的差值，默认值为1。如果指定了step，则start值会自动调整为start + step。
• dtype（可选）：序列中元素的类型，默认值为float64。可以指定为int、float等类型。

使用示例：

import numpy as np

# 生成一个从0到9的等差数列，步长为2
arr1 = np.arange(0, 10, 2)
print(arr1)  # 输出：[0 2 4 6 8]

# 生成一个从2到18的等差数列，步长为3
arr2 = np.arange(2, 19, 3)
print(arr2)  # 输出：[ 2  5  8 11 14 17]

# 生成一个从5开始，到50之前（不包括50）的等差数列，步长为5
arr3 = np.arange(5, 50, 5)
print(arr3)  # 输出：[ 5 10 15 20 25 30 35 40 45]

# 生成一个浮点数等差数列，从0到1，步长为0.1
arr4 = np.arange(0, 1, 0.1)
print(arr4)  # 输出：[0.  0.1 0.2 0.3 0.4 0.5 0.6 0.7 0.8 0.9]