numpy.insert()

2022-09-02 19:29:41 浏览数 (1)

Numpy函数

  1. insert函数 numpy.insert可以有三个参数(arr,obj,values),也可以有4个参数(arr,obj,values,axis): 第一个参数arr是一个数组,可以是一维的也可以是多维的,在arr的基础上插入元素 第二个参数obj是元素插入的位置 第三个参数values是需要插入的数值 第四个参数axis是指示在哪一个轴上对应的插入位置进行插入

如果第四个参数没有给出,那么就默认为先对arr进行flatten操作,变为一维数组,然后再在对应的位置上插入对应的值。 下面举例说明:

代码语言:javascript复制
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]

0 人点赞