python numpy.sum 参数axis，keepdims

keepdims主要用于保持矩阵的二维特性

import numpy as np
a = np.array([[1,2],[3,4]])

# 按行相加，并且保持其二维特性
print(np.sum(a, axis=1, keepdims=True))

# 按行相加，不保持其二维特性
print(np.sum(a, axis=1))

输出

array([[3], [7]])
array([3, 7])

转载 https://blog.csdn.net/u012560212/article/details/78393836

1. >>> import numpy as np  
2. >>> b=np.arange(24).reshape(2,3,4)  
3. >>> b  
4. array([[[ 0,  1,  2,  3],  
5.         [ 4,  5,  6,  7],  
6.         [ 8,  9, 10, 11]],  
7.   
8.        [[12, 13, 14, 15],  
9.         [16, 17, 18, 19],  
10.         [20, 21, 22, 23]]])  
11. >>> b.sum(axis=0)  
12. array([[12, 14, 16, 18],  
13.        [20, 22, 24, 26],  
14.        [28, 30, 32, 34]])  

1.  b.sum(axis=1)  
2. array([[12, 15, 18, 21],  
3.        [48, 51, 54, 57]])  

1.  b.sum(axis=2)  
2. array([[ 6, 22, 38],  
3.        [54, 70, 86]])  

axis = 0时：（2，3，4）2去掉，即将两个矩阵的每行每列相加例，0+12，1+13

axis = 1时：（2,3,4） 3去掉，即每个矩阵的按列相加

axis = 2时： （2,3,4）4去掉，即每个矩阵的按行相加