首先,重点是:轴就是对数组进行运算时,运算基本单位的方向(记住这个就够了)。

一般用axis = 0/1/2/-1,-1代表当前操作的数组最大的轴。一维就是0,二维就是1,三维就是2

一维只有一个轴0

二维两个0和1

三维是0,1,和2

举个例子,二维数组

如果要arr.max()会返回整个数组最大的数字,是9

如果是arr.max(axis = 0),那么基本计算单位的方向是axis = 0方向(看上面的二维数组方向图),也就是说,基本计算单位就是一列一列的,返回的是array([8,7,9,6,5])。也就是说在每个基本单位里面找最大值。

如果是arr.max(axis = 1),那么基本计算单位方向是axis = 1方向(看上面的二维数组方向图),也就是说,基本计算单位就是一行一行的,返回的是array([9,8,9,5])。也就是说在每个基本单位里面找最大值。

再举个例子,三维数组。

arr = np.array([[[1, 2],
                 [3, 4]], 
                [[5, 6],
                 [7, 8]]])

arr.sum(axis=0)
#array([[ 6,  8],
#      [10, 12]])

arr.sum(axis=1)
#array([[ 4,  6],
#       [12, 14]])

arr.sum(axis=2)
#array([[ 3,  7],
#       [11, 15]])

看上面的代码,我把数组写成这个样子,你基本上就明白了。基本计算单位方向是0轴,看上面的三维轴方向。基本单位就是(1,5),(2,6)------这样。那么相加就得到了注释的那个结果。

基本计算方向是1的时候,就是来到二维数组的领域了,基本就计算单位就是(1,3)---

基本单位方向是2的时候,以此类推。

聪明的你一定掌握了吧!   谢谢你给我点的赞奥嘿嘿

Logo

魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。

更多推荐