Numpy insert函数

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]