先讲最简单的两维数组:
axis=0,代表沿竖直方向;
axis=1,代表沿水平方向。
可以这么来记忆。
假设一个数组
A =[[a00,a01]
[a10,a11]]
数组里面的每个元素,通过2个数字表示其相对位置。例a00中的00,表示第0行,第0列的元素。
axis=0就是以这两个数字中的第一个数字为基准进行操作。
假设现在要对A数组元素进行sum
则B=A.sum(axis=0)=[a00+a10,a01+a11]。
就是a00+a10他们的第axis=0轴方向的数字是改变的,分别为0和1;其他轴数字不变都为0
多维数组同理:
A=[[[a000,a001]
[a010,a011]]
[[a100,1101]
[a110,a111]]]
B=A.sum(axis=0)=[[a000+a100,a001+a101],
[a010+a110,a011+a111]]