在没有循环的Numpy中添加子矩阵

Question

让我说我得到了这个 a = np.arange(9).reshape((3,3))我希望得到一个[9,12,15]的numpy数组，这是

的结果

[0+3+6, 1+4+7, 2+5+8]

Answer 1

您可以传递numpy.array.sum()：

来使用axis=0功能

>>> a.sum(axis=0)
array([ 9, 12, 15])

Answer 2

使用numpy.sum功能并指定您想要求和的轴，在您的情况下为0：

import numpy as np

a = np.arange(9).reshape((3,3))
a_sum = np.sum(a, axis=0)

print a_sum

这会给你：

[ 9 12 15]

Kasramvd的答案使用了面向对象的方法，有些人更喜欢这种方法：

a_sum = a.sum(axis=0)