python中某些函数axis参数的理解

首先是转载CSDN用户“虾米儿xia”的一句话，我觉得非常精炼

总结为一句话：设axis=i，则numpy沿着第i个下标变化的方向进行操作。

当然，这个i是从0开始数的，作为程序员的你一定不会搞错。

axis意为“轴”，它指定了函数在张量（矩阵、等等）上进行操作的方向。
例如有一个ndarray，名叫A，A.shape=(3,8,5,7)。
那么np.sum(A, axis=2)计算的结果的shape就是(3,8,7)。
假设这个shape是(3,8,7)的ndarray变量名为B，那么实际上：

B[i][j][k]=A[i][j][0][k]+A[i][j][1][k]+A[i][j][2][k]+A[i][j][3][k]+A[i][j][4][k]

以下代码你可以自己跑一下试试：

import numpy as np

A=np.random.randn(3,8,5,7)
print("A.shape=",A.shape)

B=np.sum(A,axis=2)
print("B.shape=",B.shape)

预期输出为：

A.shape= (3, 8, 5, 7)
B.shape= (3, 8, 7)