2D数组到3D数组的数量，扩展第三维

Question

我有 10 个不同的大小矩阵(60, 100)。我想将它们沿第三维放在 for 循环内，以便最终形状为(10, 60, 100).我尝试使用 concatenate 并最终得到 size (600, 100)。

Answer 1

numpy.stack 将允许您沿新轴连接。

arrays = [mat1,mat2,.....]

out=np.stack(arrays, axis=0)

Answer 2

另一种方式供参考：

out = np.concatenate((mat1[np.newaxis,:],mat1[np.newaxis,:],...),axis=0)

Answer 3

尝试创建一个用二维数组填充的新数组

new3DArray = numpy.empty(10, 60, 100)