python - 在python中将2D数组插入3D数组

Question

假设我制作了以下数组:

>>> a = zeros((2,2,2))
>>> b = ones((2,2))

如何将b插入到a的中间？所以我的结果数组看起来像

array([[[ 0.,  0.],
        [ 0.,  0.]],

       [[ 1.,  1.],
        [ 1.,  1.]],

       [[ 0.,  0.],
        [ 0.,  0.]]])

Answer 1

您可以使用numpy.insert() :

a = np.insert(a, 1, b, axis=0)