对于numpy包中的axis参数的理解

对于numpy包中的axis参数的理解

在numpy中，对于多维数组进行sum,mean,min,max,sort的操作时，均会涉及到axis这一参数。那么axis具体是什么呢？

我们先引入一个切片(slices)的概念:

• 如果$k$维数组$A\in {\mathbb{R}}^{n_1\times n_2\times ...\times n_k}$ ，则$A$的Slices为固定其中1个索引位置之后形成的$k-1$维数组

ok，有了slices的概念，我们如何对高维数组进行sum,mean,min,max,sort之类的运算呢？

以sum函数为例。对于一个$k$维数组$A\in {\mathbb{R}}^{n_1\times n_2\times ...\times n_k}$，在axis=i上进行运算就是在第i个维度上对相应的slices进行运算。

以sum函数在axis=i上运算为例,这个过程就相当于计算：
$$np.sum(A,axis=i)=$$