a=np.insert(arr, obj, values, axis)
#arr原始数组,可一可多,obj插入元素位置,values是插入内容,axis是按行按列插入(0:行、1:列)。
import numpy as np
# 如果axis没有给出,相当于是做降维操作,与一维数组一致
a = np.array([1,4,6,5,6,8])
b=np.insert(a,0,9)

# 输出
array([9, 1, 4, 6, 5, 6, 8])#插入元素都是在所给位置之前
a = np.array([[1,2],[3,4],[5,6]])
b=np.insert(a,1,11,axis = 1)

# 输出
array([[ 1, 11,  2],
       [ 3, 11,  4],
       [ 5, 11,  6]])

a = np.array([[1,2],[3,4],[5,6]])
b=np.insert(a,1,[2,6],axis = 0)

# 输出
array([[1, 2],
       [2, 6],
       [3, 4],
       [5, 6]])

Logo

AtomGit 是由开放原子开源基金会联合 CSDN 等生态伙伴共同推出的新一代开源与人工智能协作平台。平台坚持“开放、中立、公益”的理念,把代码托管、模型共享、数据集托管、智能体开发体验和算力服务整合在一起,为开发者提供从开发、训练到部署的一站式体验。

更多推荐