在Python中将2D数组插入3D数组中

Question

让我说我制作以下数组：

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

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

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)

Answer 2

使用tuple来显示与结果相同的方法。 代码：

b = numpy.ones((2,2))
a = numpy.zeros((2, 2))
tpl = a,
tpl = b,
tpl = a,

因此，您可以将其视为3dArray，也可以使用for循环向其中添加多数组。希望有帮助。

Answer 3

a=np.zeros((3,2,2))
b=np.ones((2,2))

a[1]=b*1

这将产生你的输出。