python中numpy.arange的用法技巧

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

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

参数说明：

• start（可选）：序列的起始值，默认值为0。
• stop（必需）：序列的结束值（不包括该值），必须是整数。
• step（可选）：序列中相邻元素之间的差值，默认值为1。如果为负数，则生成递减序列。
• dtype（可选）：输出数据的类型，默认值为float64。可以指定为其他数值类型，如int32、float32等。

用法技巧：

1. 生成0到10之间的整数序列：

import numpy as np seq = np.arange(11) print(seq)  # 输出：[0 1 2 3 4 5 6 7 8 9 10] 

2. 生成5到-5之间的整数序列，步长为2：

import numpy as np seq = np.arange(5, -5, -2) print(seq)  # 输出：[ 5  3 -1 -3 -5] 

3. 生成浮点数序列，起始值为0.0，结束值为1.0，步长为0.1：

import numpy as np seq = np.arange(0.0, 1.0, 0.1) print(seq)  # 输出：[0.  0.1 0.2 0.3 0.4 0.5 0.6 0.7 0.8 0.9] 

4. 生成指定类型的序列：

import numpy as np seq = np.arange(0, 10, dtype=np.int32) print(seq)  # 输出：[0 1 2 3 4 5 6 7 8 9]