Python函数:np.sum()以及axis=0、axis=1用法
常用于矩阵求和计算,以下用法分为三种情况来介绍!
格式: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]
numpy.argmin(a, axis=None, out=None)[source]
给出axis方向最小值的下标
>>> a = np.arange(6).reshape(2,3) >>> a array([[0, 1, 2], [3, 4, 5]]) >>> np.argmin(a) 0 >>> np.argmin(a, axis=0) array([0, 0, 0]) >>> np.argmin(a, axis=1) array([0, 0])
Parameters:
a : Input array.
axis : 默认将输入数组展平。否则,按照axis方向
out : 可选
Returns:
index_array : 下标组成的数组。shape与输入数组a去掉axis的维度相同。