numpy中的各种函数细节点之numpy.sum()
numpy.sum(a,axis)
a:需要求和的Elements
axis:按行或者列求和,0表示列,1表示行
用法
>>> np.sum([[0, 1], [0, 5]], axis=0)
array([0, 6])
>>> np.sum([[0, 1], [0, 5]], axis=1)
array([1, 5])
重点来了
如果你传进去的是一个二维数组,在经过numpy.sum()之后,会返回a.ndim - 1维。比如说上面的[[0, 1], [0, 5]]是
原创
2021-04-10 21:01:52 ·
351 阅读 ·
0 评论