假设我有N个numpy数组,每个数组的大小分别为(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数组?
您可以沿第一个轴将它们串联起来。如果您的3个numpy数组分别命名为x1,x2和x3,则新数组将定义为x_combined = np.concatenate((x1,x2,x3),axis=0)
尝试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)