如何向 Numpy 数组添加新维度？

Question

我从一个图像的 numpy 数组开始。

 In[1]:img = cv2.imread('test.jpg')

该形状是您对 640x480 RGB 图像所期望的形状。

 In[2]:img.shape
Out[2]: (480, 640, 3)

但是，我拥有的这张图片是一帧视频，长度为 100 帧。理想情况下，我希望有一个包含该视频中所有数据的数组，这样 img.shape 返回 (480, 640, 3, 100) 。

将下一帧（即下一组图像数据，另一个 480 x 640 x 3 阵列）添加到我的初始阵列的最佳方法是什么？

原文由 Chris 发布，翻译遵循 CC BY-SA 4.0 许可协议

Answer 1

可以将维度添加到 numpy 数组，如下所示：

 image = image[..., np.newaxis]

原文由 abcd 发布，翻译遵循 CC BY-SA 4.0 许可协议