numpy之sum()
- 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
- 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])
>>>
- 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