A-A+
Python NumPy中的数组求和函数sum
在Python中可以使用NumPy中的sum函数来进行数值求和,sum方法的格式如下:
numpy.sum(a, axis=None, dtype=None, out=None, keepdims=, initial=)
常用参数含义如下:
a - 待求和的数组;
axis - 求和的方向,None或没有axis参数表示全部数据相加,axis=0,表示按列相加,axis=1,表示按行相加,对于一维数组,只有axis=0,表示按行相加,没有axis=1,输入axis=1会报错;
dtype - 输出结果数据的数据类型。
示例代码:
- >>> import numpy as np
- >>> A=np.array([[1,2],[3,4],[5,6]])
- >>> np.sum(A)
- 21
- >>> np.sum(A,axis=0)
- array([ 9, 12])
- >>> np.sum(A,axis=1)
- array([ 3, 7, 11])