python - 将 1D 数组连接到 3D 数组

Question

我有一个三维数组A，形状为(5774,15,100)，还有一个一维数组B，形状为 (5774,)。我想添加这些以获得形状为 (5774,15,101) 的另一个矩阵 C。

我正在使用 hstack 作为

C = hstack((A ,np.array(B)[:,None]))

我收到以下错误，任何建议。

ValueError: could not broadcast input array from shape (5774,15,100) into shape (5774)

Answer 1

您需要使用 np.concatenate(它可以连接不同形状的数组，这与各种 np.*stack 方法不同)。然后，您需要使用 np.broadcast_to 将 (5774,) 形数组转换为 (5774, 15, 1)(因为 concatenate 仍然需要所有数组具有相同的维数)。

C = np.concatenate((A, 
                    np.broadcast_to(np.array(B)[:, None, None], A.shape[:-1] + (1,))), 
                   axis = -1)

检查:

A = np.random.rand(5774, 15, 100)
B = np.random.rand(5774)
C = np.concatenate((A, 
                    np.broadcast_to(np.array(B)[:, None, None], A.shape[:-1] + (1,))), 
                   axis = -1)
C.shape

Out: (5774, 15, 101)