numpy arrays 到一個 numpy 陣列的列表

Question

假設我有 N numpy arrays，每個大小為 (x, y, z)，其中 y 和 z 對所有人都相同，但 x 每個人不同。 我如何將這些組合到大小為 (w, y, z) 的 numpy 數組中，其中 w 是所有 x 的總和。

或者，對於一個數字示例：我有 3 個 numpy 數組的列表，大小為 (14, 32, 32), (7, 32, 32), (50, 32, 32)。 如何有效地將它們變成（71、32、32）大小的 numpy 陣列？

Answer 1

您可以將它們沿第一個軸連接起來。 如果您的 3 numpy arrays 被命名為 x1、x2 和 x3，則您的新數組將定義為x_combined = np.concatenate((x1,x2,x3),axis=0)

Answer 2

嘗試np.vstack

a, b , c = np.ones((14, 32, 32)), np.ones((7, 32, 32)), np.ones((50, 32, 32))

out = np.vstack([a,b,c])

In [119]: a.shape
Out[119]: (14, 32, 32)

In [120]: b.shape
Out[120]: (7, 32, 32)

In [121]: c.shape
Out[121]: (50, 32, 32)

In [122]: out.shape
Out[122]: (71, 32, 32)