有时候,需要用到固定小数步长的某一区间内的排列,如[0, 0.2, 0.4, 0.6],此时,由于常用的range()函数只支持固定整数步长,所以就无法通过range()函数来实现。
这种情况下,就可以通过numpy.arange()函数实现,具体如下:
numpy.arange()
- 返回值:返回一个有起点和终点的可固定小数步长的排列;
- 参数:
- 一个参数时,参数值为终点,起点取默认值0,步长取默认值1;
- 两个参数时,第一个参数值为起点,第二个参数值为终点,步长取默认值1;
- 三个参数时,第一个参数为起点,第二个参数为终点,第三个参数为步长(支持小数)。
实例:
import numpy as np
## 一个参数
a = np.arange(3)
print(a) ## [0 1 2]
## 两个参数
b = np.arange(4,8)
print(b) ## [4 5 6 7]
## 三个参数
c = np.arange(0,1,0.2)
print(c) ## [0. 0.2 0.4 0.6 0.8]