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]])