Python函数:np.sum()以及axis=0、axis=1用法

Python基本函数:np.sum()

        常用于矩阵求和计算,以下用法分为三种情况来介绍!
格式:np.sum(a)
           np.sum(a, axis=0) ------->列求和
           np.sum(a, axis=1) ------->行求和
注意:特别注意后两种用法。

一、具体用法

In :       	a = np.array([[0, 2, 1], [3, 5, 6], [0, 1, 1]]) #得到一个矩阵a
Out:      	[[0 2 1]
 			 [3 5 6]
 			 [0 1 1]]     									
                                             				
In :       	a.sum()                    						#将a中所有元素求和 0+2+1+3+5+6+0+1+1=19
Out: 		19
 
In : 		np.sum(a,axis=1)     							# 求出每行元素的和
Out: 		[ 3  14  2]                          			

In : 		np.sum(a,axis=0)								# 求出每列元素的和
Out: 		[3  8  8]                         				
发布了36 篇原创文章 · 获赞 5 · 访问量 3万+

猜你喜欢

转载自blog.csdn.net/qq_40520596/article/details/102738481