numpy insert

import numpy as np

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

print(a)

[[1 2]
 [3 4]
 [5 6]]

print(np.insert(a,3,[11,12]))

[ 1  2  3 11 12  4  5  6]

在第三个元素后面添加11,12并且被展平

print(np.insert(a,3,[11,12],axis=0)) (原数组,层数,轴数)

[[ 1  2]
 [ 3  4]
 [ 5  6]
 [11 12]]

在第3层的末尾添加

print(np.insert(a,1,[11],axis=0))

[[ 1  2]
 [11 11]
 [ 3  4]
 [ 5  6]]

在第一层的末尾添加并且自动变成和整体相同的层数

print(np.insert(a,1,11,axis=1))

在每层的第一个元素后面添加11 第三个参数是否为列表并不重要 重要的是axis为1