假設我有 N 個 numpy 數(shù)組，每個數(shù)組的大小為 (x, y, z)，其中 y 和 z 對于所有數(shù)組都是相同的，但每個數(shù)組的 x 不同。我如何將這些組合成一個大小為 (w, y, z) 的 numpy 數(shù)組，其中 w 是所有 x 的總和?；蛘?，對于一個數(shù)字示例：我有一個包含 3 個大小為 (14、32、32)、(7、32、32)、(50、32、32) 的 numpy 數(shù)組的列表。如何有效地將這些轉(zhuǎn)換為 (71, 32, 32) 大小的 numpy 數(shù)組？
2 回答
胡說叔叔
TA貢獻1804條經(jīng)驗 獲得超8個贊
您可以將它們沿第一個軸連接起來。如果您的 3 個 numpy 數(shù)組被命名為 x1、x2 和 x3，那么您的新數(shù)組將被定義為x_combined = np.concatenate((x1,x2,x3),axis=0)
浮云間
TA貢獻1829條經(jīng)驗 獲得超4個贊
嘗試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)
添加回答
舉報
0/150
提交
取消