numpy中的sum函数

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/weixin_40244153/article/details/79714477

Numpy中的sum函数

函数无参数时,将所有元素相加,结果为一个数;

参数:axis=0,按列相加,结果为一维数组;

参数:axis=1,按行相加,结果为一维数组

>>> from numpy import *
>>> a=array([[1,2,3],[4,5,5]])
>>> a.sum()
20
>>> a.sum(axis=0)
array([5, 7, 8])
>>> a.sum(axis=1)
array([ 6, 14])

猜你喜欢

转载自blog.csdn.net/weixin_40244153/article/details/79714477