如何对嵌套数组中的数组求和？答案

【问题标题】：How to sum arrays in nested arrays?如何对嵌套数组中的数组求和？
【发布时间】：2021-10-10 22:17:21
【问题描述】：

我有一个嵌套数组

array([[1,2,4], [2,5,6]])

我想将其中的每个数组相加得到：

array([[7], [13]])

如何做到这一点？当我做np.array([[1,2,4], [2,5,6]]) 它给了

array([7, 13])

【问题讨论】：

你知道如何给数组添加维度吗？ np.sum(arr, axis=1)[:,None]。 keep_dims 参数很方便，但你也应该熟悉这个None 成语。

【解决方案1】：

在轴 1 上使用 sum：

>>> a = np.array([[1,2,4], [2,5,6]])
>>> a.sum(axis=1, keepdims=True)
[[ 7]
 [13]]

或者没有 numpy：

>>> a = [[1,2,4], [2,5,6]]
>>> [[sum(l)] for l in a]
[[7], [13]]

【讨论】：

【解决方案2】：

我不确定array() 函数是什么，但如果它只是一个列表，那么这应该工作：

a=array([[1,2,4], [2,5,6]])
b=[[sum(x)] for x in a] #new list of answers

【讨论】：