python - 如何向 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 数组)添加到我的初始数组的最佳方法是什么？

Answer 1

一个维度可以添加到一个numpy数组中，如下所示:

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