Python——Numpy中的sum

今天做运算的时候需要计算一个数组中所有元素的和，网上有很多用np.sum()，但是都是需要设定行列进行分步计算。

其实可以直接用np.sum()进行运算

测试代码如下：

import numpy as np
a = [[1,2,3] ,[4, 5, 6], [7, 8, 9]]
b = np.sum(a,axis=0)
c = np.sum(b,axis=0)
aa = np.sum(a)
print(aa,b,c)

结果：

45 [12 15 18] 45

可以看到直接运算和分步计算的结果是一样的