numpy.insert()

Numpy函数

  1. 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]

承接Matlab、Python和C++的编程,机器学习、计算机视觉的理论实现及辅导,本科和硕士的均可,咸鱼交易,专业回答请走知乎,详谈请联系QQ号757160542,非诚勿扰。

猜你喜欢

转载自blog.csdn.net/weixin_36670529/article/details/114055335