【发布时间】:2019-02-21 04:36:21
【问题描述】:
我想知道如何通过循环获取 3x3 NumPy 数组中的值总和。
例如:
[[1,2,3],[4,5,6],[7,8,9]]
输出应该是[6,15,24]。
【问题讨论】:
我想知道如何通过循环获取 3x3 NumPy 数组中的值总和。
例如:
[[1,2,3],[4,5,6],[7,8,9]]
输出应该是[6,15,24]。
【问题讨论】:
尝试使用 np.array.sum 和 axis 参数作为一个:
arr.sum(1)
例子:
>>> arr=np.array([[1,2,3],[4,5,6],[7,8,9]])
>>> arr.sum(1)
array([ 6, 15, 24])
或者如果是列表:
np.sum(array,1)
例子:
>>> arr=[[1,2,3],[4,5,6],[7,8,9]]
>>> np.sum(arr,1)
array([ 6, 15, 24])
>>>
【讨论】: