numpy之arange函数

第一次在CSDN上写东西,就当是自己编程学习上的记录吧,从最简单的写起。参考链接

np.arange(start, stop, step, dtype)

start:起始值,可忽略不写,默认从0开始;
stop:结束值,生成的元素不包括结束值;
step:步长,可忽略不写,默认为1;
dtype:设置显示元素的数据类型,默认为None。

arange函数用于创建等差数组,跟range函数类似,不同之处有两个:
1.使用前,arange需要引入numpy。
2. arange返回一个数据而不是list。

对应的代码示例如下:

1.import numpy as np

2.使用不同参数做对比

  • 输入一个参数
>>> a=np.arange(5)
[0 1 2 3 4]
  • 输入两个参数
>>> b1=np.arange(1,5)
[1 2 3 4]

>>> b2=np.arange(2,5)
[2 3 4]
  • 输入三个参数
>>> c=np.arange(1,10,2)
[1 3 5 7 9]

还可以与reshape连用

可以对生成的等差一维数组,进行重塑,使用的是 .reshape(行,列),示例:

>>> b1.reshape(2,2)
[[1 2]
 [3 4]]
#注意:对数组重塑后的元素个数不能大于原数组本身的元素个数,不然就会报错
#比如说,b1生成了4个元素,reshape成(2,3)就是6个元素,元素个数不一致,会报错

你可能感兴趣的:(python,numpy)