求和函数sum()对比分析——Python & Matlab

数组求和有三种场景：数组所有元素求和、数组按列求和、数组按行求和。

① 数组所有元素求和；

Matlab代码：

%A为二维数组
sum(A(:))
sum(sum(A))

Python代码:

#A为二维数组
numpy.sum(A) #numpy的sum()函数，默认求数组所有元素的和
sum(sum(A))  #自带的sum()函数

② 数组按列求和；

Matlab代码：

%A为二维数组
sum(A) %默认按列求和
sum(A,1) %参数设置为1，按列求和

Python代码：

#A为二维数组
numpy.sum(A,axis=0) #numpy的sum()函数,设置参数按列求和
sum(A)  #自带的sum()函数，默认按列求和

③ 数组按行求和

Matlab代码：

%A为二维数组
sum(A,2) %参数设置为2，按行求和

Python代码:

#A为二维数组
numpy.sum(A,axis=1) #numpy的sum()函数,设置参数axis=1，按行求和

因此，为了避免搞混，记住其中一套即可：

	Matlab	Python
所有元素求和	sum(A(:))	numpy.sum(A)
按行求和	sum(A,2)	numpy.sum(A,axis=1)
按列求和	sum(A,1)	numpy.sum(A,axis=0)

注：养成设置函数参数的习惯，这样有利于加深对函数的理解，也更不容易出错。​​​​​​​