如何添加列表元素

Question

我有一个列表（Y），其中包含不同的 numpy 数组 length.The 列表包含超过 50 个元素（我只拿了五个进行测试）。列表的形状是 5，我可以打印每个元素的形状如下所示，输出在评论中

print(Y.shape) #(5,)
print(Y[0].shape) #(600, 2)
print(Y[1].shape) #(250, 2)
print(Y[2].shape) #(300, 2)
print(Y[3].shape) #(200, 2)
print(Y[4].shape) #(100, 2)

列表中的每个元素都有不同的长度 (600,250,300,200,100) 但都有 [1 0] 或 [0 1] dimension.I 想要添加这些元素并获得输出

(1450,2)

我试过了

Y=np.sum(Y, axis=0)

它给出了广播错误，ValueError：操作数无法与形状一起广播 (600,2) (250,2) 我知道这需要 (600,2) (600,2) 或 (250, 2) (250,2) 但我想添加 600 和 250。

同样的函数适用于像 [80,20,30] 和 [40,20,30] 这样的三维数组，我得到的输出是 [120,20,30]

我怎样才能add/sum这些元素？

Answer 1

使用np.concatenate:

np.concatenate(Y, axis=0)

例如：

import numpy as np

Y1 = np.ones((100, 2))
Y2 = np.ones((200, 2))
Y3 = np.ones((300, 2))

np.concatenate([Y1, Y2, Y3], axis=0).shape   # (600, 2)

如何添加列表元素

how to add elements of a list

python

arrays

numpy

array-broadcasting