numpy数组到一个numpy数组的列表

问题描述 投票:1回答:2

假设我有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数组?

python numpy numpy-ndarray
2个回答
2
投票

您可以沿第一个轴将它们串联起来。如果您的3个numpy数组分别命名为x1,x2和x3,则新数组将定义为x_combined = np.concatenate((x1,x2,x3),axis=0)


1
投票

尝试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)
© www.soinside.com 2019 - 2024. All rights reserved.