python - 关于Python NumPy二维数组的问题

Question

我有一个大小为[49152, 3]的 NumPy 数组，我想附加另一个大小相同的数组[49152, 3]作为新行成为[2, 49152, 3]。我可以知道这样做的方法吗？所有这些都是来自 192x256 分辨率图像的 RGB 图像像素值。

标签： pythonnumpy

Answer 1

您很可能需要使用numpy.stack：

>>> a = np.random.rand(49152, 3)
>>> b = np.random.rand(49152, 3)

>>> np.stack([a, b]).shape
(2, 49152, 3)

您可以提供相同形状的数组列表（或元组np.stack），并将它们堆叠在新创建的轴上（默认情况下在第一个轴上：）axis=0。