版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
import numpy as np
a = np.array([[0,3],[1,4]])
#a.shape是（2,2）
#axis=1表示按行相加
#keepdims保持其二维特性
print(np.sum(a, axis=1, keepdims=True))
# 按行相加，不保持其二维特性
print(np.sum(a, axis=1))
输出为
array([[3], [5]])，shape是（2,1）
array([3, 5])，shape是（1,2）