我有一个类似的列表:
rhs
并且我可以保证每个子列表的大小相同。我需要对每个位置的每个元素求和(产生一个与子列表大小相同的列表)。
这是这样生成的:
[array([ 0.6, 0.6, -0.6, -0.6, -0.6, -0.6, -0.6, -0.6, -0.6, -0.6]), array([ 0.35, 0.35, 0.35, 0.35, 0.35, 0.35, 0.35, 0.35, -0.35,
-0.35])]
如果有一种方法可以在不提前知道子列表大小的情况下立即建立总和,那就更好了。
答案 0 :(得分:1)
np.sum(sums, axis=0)
您甚至不需要创建新的2D numpy数组。