【转载】关于numpy.arrage()

#转载于宁静致远*原文链接https://blog.youkuaiyun.com/weixin_40522801/article/details/106467932

关于numpy.arrage()

Numpy中arange() 主要是用于生成数组，用法如下：

函数返回一个有终点和起点的固定步长的排列，如[1,2,3,4,5]，起点是1，终点是5，步长为1。

参数个数情况： np.arange()函数分为一个参数，两个参数，三个参数三种情况

1. 一个参数时，参数值为终点，起点取默认值0，步长取默认值1。
2. 两个参数时，第一个参数为起点，第二个参数为终点，步长取默认值1。
3. 三个参数时，第一个参数为起点，第二个参数为终点，第三个参数为步长。其中步长支持小数

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

例子：

# 注意：浮点数参数和整数参数结果不太一样，整数不会包含最后一个，而浮点数会包含

from numpy import *  # 引入numpy

A = arange(5)  # 只有结束项

print(A)  # 结果 [0 1 2 3 4] 结果不包含结束项

print(type(A))  # 结果 <class 'numpy.ndarray'>

A = arange(1, 5)  # 起点为1，步长默认为1

print(A)  # 结果 [1 2 3 4]

A = arange(1, 5, 2)  # 步长默认为2

print(A)  # 结果 [1 3]

A = arange(1, 5.2, 0.6)  # 浮点数参数，结果就不一定完全符合了

print(A)  # 结果 [1.  1.6 2.2 2.8 3.4 4.  4.6 5.2]