關於Python NumPy二維數組的問題

Question

我有一個大小為[49152, 3]的 NumPy 數組，我想將 append 另一個大小相同的數組[49152, 3]作為新行成為[2, 49152, 3] 。 我可以知道這樣做的方法嗎？ 所有這些都是來自 192x256 分辨率圖像的 RGB 圖像像素值。

Answer 1

您可以在 Python 中使用numpy的insert function。

x = np.array([49152, 3])    
X = np.insert(x, 0, 2)

x 是 append 的原始數組到 0 是您需要 append 的索引 2 是需要附加的。

Answer 2

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

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

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

您可以提供相同形狀的 arrays 的列表（或元組），並且np.stack會將它們堆疊在新創建的軸上（默認情況下在第一個軸上： axis=0 ）。