numpy常用函数之arange函数
2018-10-09 本文已影响0人
郭祺迦
2、np.arange([start, ]stop, [step, ]dtype=None)
作用:
arange函数用于创建等差数组
start:可忽略不写,默认从0开始;起始值
stop:结束值;生成的元素不包括结束值
step:可忽略不写,默认步长为1;步长
dtype:默认为None,设置显示元素的数据类型
举例:
import numpy as np
nd1 = np.arange(5)#array([0, 1, 2, 3, 4])
nd2 = np.arange(1,5)#array([1, 2, 3, 4])
nd3 = np.arange(1,5,2)#nd3 = np.arange(1,5,2)
可以对生成的等差一维数组,进行重塑,使用的是reshape(行,列)
nd2.reshape(2,2)#array([[1, 2], [3, 4]])
注意:对数组重塑后的元素个数不能大于原来本身的元素个数,不然会报错
比如说,nd2生成了四个元素,你要重塑(2,3)就是六个元素,会报错的