看起来挺简单的样子,但是在给sum函数中加入参数。sum(a,axis=0)或者是.sum(axis=1) 就有点不解了

在我实验以后发现 我们平时用的sum应该是默认的axis=0 就是普通的相加 ,而当加入axis=1以后就是将一个矩阵的每一行向量相加

例如:

输入是一个两行三列的矩阵[[0,1,2]

                                             [2,1,3]]

然后设置axis=1,则计算每一行的向量之和,即为[3,6]

 

下面对二维数组进行更多的实验:

a.sum()是将矩阵中所有的元素进行求和

a.sum(axis = 0)是普通的相加,即对应列相加

 

下面对一维数组进行更多的实验:

可以看到,对于一维数组,a.sum()仍然是将矩阵中所有的元素进行求和,a.sum(axis=0)的效果与a.sum()的效果完全一致

而a.sum(axis = 1)则会报错,是因为一维数组,只有第0轴,没有第1轴

 

更多实验:


本文转载:CSDN博客