二维二维numpy数组的组合列表

Question

如何结合N，二维（尺寸为R x C的numpy数组来创建形状为(N, R, C)的3D numpy数组？现在，N-2D numpy数组包含在列表中，我希望它成为3D numpy数组。假设X是我的2D numpy数组列表，如果我只做np.array(X)，就会得到形状(N,)的东西。如果我做np.vstack(X)，我会得到(N x R, C)的形状。我该如何解决这个问题？

Answer 1

您可以使用np.stack：

test = np.stack([np.ones([2, 3]) for _ in range(4)]) 
print(test.shape) # (4, 2, 3)

Answer 2

您可以使用：

np.array([np.array(x) for x in ArrayList])