numpy arange

numpy arange

简介

在进行科学计算和数据处理时，经常会用到生成一组连续的数列。而numpy中的arange函数就是用来生成一个指定范围内的等间隔的数列的。它的用法简单灵活，非常方便实用，是numpy中常用的一个函数之一。

语法

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

• start：数列的起始值，默认为0
• stop：数列的结束值，生成的数列不包括该值
• step：数列的步长，默认为1
• dtype：输出数组的数据类型，默认为None

示例

生成一个从0到9的数列

import numpy as np

arr = np.arange(10)
print(arr)

输出：

[0 1 2 3 4 5 6 7 8 9]

生成一个从3到10的数列

import numpy as np

arr = np.arange(3, 11)
print(arr)

输出：

[ 3  4  5  6  7  8  9 10]

生成一个从0到10，步长为2的数列

import numpy as np

arr = np.arange(0, 11, 2)
print(arr)

输出：

[ 0  2  4  6  8 10]

生成一个从0到1，步长为0.2的数列

import numpy as np

arr = np.arange(0, 1, 0.2)
print(arr)

输出：

[0.  0.2 0.4 0.6 0.8]

注意事项

1. arange生成的数列不包括stop的值，即生成的数列范围是[start, stop)
2. 步长可以是小数，可以为负数，但要注意控制好生成的范围，避免出现意外情况
3. 当未指定dtype时，根据生成的数列类型自动确定输出数组的数据类型

通过上面的示例，我们可以看出numpy.arange函数的用法非常简单，并且可以根据需求生成不同范围和步长的数列，非常灵活便捷。在实际应用中，经常会用到该函数来生成需要的数列，为接下来的数据处理和分析提供基础。