numpy.sum()函数

import numpy as np
 
a = np.array([[1, 2, 1], 
              [3, 4, 5]])
 
# axis=0代表将行进行求和， 
# 也就是将各个列的不同行进行求和
print("axis=0:", np.sum(a, axis=0))
 
# axis=1代表将列进行求和， 
# 也就是将各个行的不同列进行求和
print("axis=1:", np.sum(a, axis=1))
 
# 所有元素求和
print(np.sum(a))
 
结果为：
axis=0: [4 6 6]
axis=1: [ 4 12]
16