如何附加差异 dim' 数组？

Question

我做了一些numpy数组np3

np1 = np.array(range(2*3*5))
np3 = np1.reshape(2,3,5)

CCD_ 2的形状是这样的：

[[[ 0  1  2  3  4]
[ 5  6  7  8  9]
[10 11 12 13 14]]
[[15 16 17 18 19]
[20 21 22 23 24]
[25 26 27 28 29]]]

然后，我制作了新的numpy阵列np_55

np_55 = np.full((3,1),55)

CCD_ 4的形状是这样的：

[[55]
[55]
[55]]

我想使用numpy数组np3和np_55制作如下的numpy数组(我称之为"答案"(：

[[[ 0  1  2  3  4 55]
[ 5  6  7  8  9 55]
[10 11 12 13 14 55]]
[[15 16 17 18 19 55]
[20 21 22 23 24 55]
[25 26 27 28 29 55]]]

但我不能同时使用numpy数组CCD_ 7和CCD_。当然，我可以制作这样的硬代码：

a = np.append((np3[0]), np3_55, axis=1)
b = np.append((np3[1]), np3_55, axis=1) 
a = a.reshape(1,3,6)
b = b.reshape(1,3,6)
np.append(a, b, axis=0)

但我不知道如何简单地解决答案。

Answer 1

您可以尝试以下操作：

import numpy as np
a = np.arange(2*3*5).reshape(2, 3, 5)
b = np.full((3,1),55)
np.c_[a, np.broadcast_to(b, (a.shape[0], *b.shape))]

它给出：

array([[[ 0,  1,  2,  3,  4, 55],
[ 5,  6,  7,  8,  9, 55],
[10, 11, 12, 13, 14, 55]],
[[15, 16, 17, 18, 19, 55],
[20, 21, 22, 23, 24, 55],
[25, 26, 27, 28, 29, 55]]])