Python中用max()、sum()等函数处理矩阵时，axis参数的作用

bigmidhu

于 2020-10-05 16:45:42 发布

阅读量1.3k

点赞数 1

本文链接：https://blog.youkuaiyun.com/weixin_44153656/article/details/108930109

版权

本文介绍了numpy库中对矩阵进行最大值提取和求和的操作，如`np.max()`和`np.sum()`函数，以及如何沿指定轴进行计算。同时，详细阐述了Python中的广播机制，解释了不同维度矩阵间的加减乘除运算，特别是向量与矩阵相除的情况，展示了numpy如何通过广播机制实现这种运算。这对于理解和应用numpy进行数值计算至关重要。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

a = array([[ 1.,  2.,  3.,  4.],
       [ 5.,  6.,  7.,  8.],
       [ 9., 10., 11., 12.]])
np.max(a, axis=1) #取每行的最大值
>>>array([ 4.,  8., 12.])
np.max(a, axis=0) #取每列的最大值
>>>array([ 9., 10., 11., 12.])

sum()

a = np.array([[ 1.,  2.,  3.,  4.],
       [ 5.,  6.,  7.,  8.],
       [ 9., 10., 11., 12.]])
cal = a.sum(axis=0) #每列各自求和 cal = [15,18,21,24]
a/cal.reshape(1,4) #a是3*4维，cal是1*4维，a每列除以cal那列的值

python中矩阵与向量的加减乘除运算（broadcasting（广播）机制）
若与矩阵参与运算的对象是一个行/列维度相同，另一个维度为1的向量，则不同的维度将按复制法展开（将低维的那个对象展开至高维），然后逐个元素加减乘除（注意乘法也可以）
所以，上面的那种不同维度矩阵相除，包括向量和常数运算都是可以的
乘法还有矩阵乘法，区别于上方