numpy之sum()

numpy之sum()

1. sum(a)默认为对输入参数中的所有元素进行求和

>>> a
array([ 0,  1,  2,  3,  4,  5,  6,  7,  8,  9, 10, 11])
>>> np.sum(a)
66


>>> b=np.arange(12)
>>> b=b.reshape(2,6)
>>> b
array([[ 0,  1,  2,  3,  4,  5],
       [ 6,  7,  8,  9, 10, 11]])
>>> np.sum(b)
66

2. sum()输入参数带有axis时，将按照指定axis进行对应求和

>>> b=np.arange(12)
>>> b=b.reshape(2,6)
>>> b
array([[ 0,  1,  2,  3,  4,  5],
       [ 6,  7,  8,  9, 10, 11]])
>>> np.sum(b)	#默认对所有元素进行求和
66
>>> np.sum(b,axis=0)		#在第一个轴展开方向上求和
array([ 6,  8, 10, 12, 14, 16])
>>> np.sum(b,axis=1)
array([15, 51])
>>> 

3. sum()输入参数axis为多个轴时，则依次按要求在axis上进行多次求和

>>> a=np.arange(12).reshape(2,2,3)
>>> a
array([[[ 0,  1,  2],
        [ 3,  4,  5]],

       [[ 6,  7,  8],
        [ 9, 10, 11]]])

>>> b=np.sum(a,axis=(0,1))	#分别在axis=0 和 1两个方向上进行求和
>>> b
array([18, 22, 26])
>>> b=np.sum(a,axis=(1,2))
>>> b
array([15, 51])

>>> b=np.sum(a,axis=(0,1,2))	#由于a为3为矩阵，所有在三个axis上分别求和相当于对所有元素进行求和 
>>> b
66