numpy进阶应用--求和

记录下一些 numpy 的进阶使用

numpy.sum()根据轴求和

和上例一样,有一个4×2的数组:

1
2
3
4
5
6
7
8
In [1]: import numpy as np
In [2]: a = np.array([[1, 2, -2, 10], [-3, 11, 99, -9]])
In [3]: a
Out[3]:
array([[ 1, 2, -2, 10],
[-3, 11, 99, -9]])

不带任何参数的np.sum()与 Python 的内置函数 sum()一样对数组的所有元素求和:

1
2
In [4]: np.sum(a)
Out[4]: 109

比普通的sum()强大的是,numpysum()可以根据数组的轴向分别进行求和,对列求和加入参数axis=0即可:

1
2
In [5]: np.sum(a, axis=0)
Out[5]: array([-2, 13, 97, 1])

对行求和时axis=1:

1
2
In [6]: np.sum(a, axis=1)
Out[6]: array([11, 98])