【Python】sum(1, keepdim=True)的理解

axis的理解
axis=0:可以理解为按照行顺序,对列进行操作
axis=1：可以理解为按照列的顺序，对行进行操作

sum(1, keepdim=True)的理解
求数组每一行的和
keepdims=True 用途：保持原数组的维度

import numpy as np
x=np.array([[0,1],[1,1],[1,2]])
x.sum(1)

运行结果

array([1, 2, 3])

x.sum(1,keepdims=True)

运行结果

array([[1],
       [2],
       [3]])