Numpy函数 insert函数 numpy.insert可以有三个参数(arr,obj,values),也可以有4个参数(arr,obj,values,axis): 第一个参数arr是一个数组,可以是一维的也可以是多维的,在arr的基础上插入元素 第二个参数obj是元素插入的位置 第三个参数values是需要插入的数值 第四个参数axis是指示在哪一个轴上对应的插入位置进行插入 如果第四个参数没有给出,那么就默认为先对arr进行flatten操作,变为一维数组,然后再在对应的位置上插入对应的值。 下面举例说明: arr2=np.array([1,2,3,4,5]) arr3=arr2.reshape(-1,1) arr=np.insert(arr3,1,[[0],[0]],axis=0) #arr=np.insert(arr3,slice(1,2),[[0],[0]],axis=0) 此处slice是切片的意思,与 numpy中切片的意思一致 print(arr3) print(arr) arr.flatten() [[1] [2] [3] [4] [5]] [[1] [0] [0] [2] [3] [4] [5]] array([1, 0, 0, 2, 3, 4, 5]) arr=np.insert(arr,4,[[0],[0]],axis=0)#此处4表示位置即obj,在第几个位置之后插入value arr=np.insert(arr,7,[[0],[0]],axis=0) arr=np.insert(arr,10,[[0],[0]],axis=0) print(arr.flatten()) [1 0 0 2 0 0 3 0 0 4 0 0 5] |
|
来自: 王野yvvrnyam9s > 《待分类》