我们以三维数组为例说明按轴计算参数的用法
定义三维数组a
a是一个(2,2,3)的三维数组
axis参数相对应的值其实是a数组的三个维度的下标(索引)
我们使用sum函数计算每组的和
当 axis = 0的时候计算方向为第0个维度,取值范围0,1
两个数,即参与求和的元素索引a[0][x][y]+a[1][x][y]
x取值范围0,1 ; y取值范围0,1,2
在不考虑 sxis = 0的维度时 [x][y]的组合有:
x=0 : [0][0],[0][1],[0][2]
x=1 : [1][0],[1][1],[1][2]
最终计算过程如下:
当 axis = 1的时候计算方向为第1个维度,取值范围0,1
两个数,即参与求和的元素索引a[x][0][y]+a[x][1][y]
x取值范围0,1 ; y取值范围0,1,2
在不考虑 sxis = 1的维度时 [x][y]的组合有:
x=0 : [0][0],[0][1],[0][2]
x=1 : [1][0],[1][1],[1][2]
最终计算过程如下:
当 axis = 2的时候计算方向为第2个维度,取值范围0,1,2
三个数,即参与求和的元素索引a[x][y][0]+a[x][y][1]+a[x][y][3]
x取值范围0,1 ; y取值范围0,1
在不考虑 sxis = 2的维度时 [x][y]的组合有:
x=0 : [0][0],[0][1]
x=1 : [1][0],[1][1]
最终计算过程如下: