NumPy的轴参数如何工作？

Question

有人可以确切解释axisNumPy中的参数做什么吗？

我很困惑。

我正在尝试使用该功能 myArray.sum(axis=num)

最初，我认为如果数组本身是3维，axis=0将返回三个元素，该元素由同一位置上所有嵌套项的总和组成。如果每个维度都包含五个维度，则我希望axis=1返回五个项目的结果，依此类推。

但是事实并非如此，文档也不能很好地帮助我（他们使用3x3x3数组，因此很难判断发生了什么）

这是我所做的：

>>> e
array([[[1, 0],
        [0, 0]],

       [[1, 1],
        [1, 0]],

       [[1, 0],
        [0, 1]]])
>>> e.sum(axis = 0)
array([[3, 1],
       [1, 1]])
>>> e.sum(axis=1)
array([[1, 0],
       [2, 1],
       [1, 1]])
>>> e.sum(axis=2)
array([[1, 0],
       [2, 1],
       [1, 1]])
>>>

显然，结果并不直观。

Answer 1

显然，

e.shape == (3, 2, 2)

轴上的总和是归约运算，因此指定的轴消失。因此，

e.sum(axis=0).shape == (2, 2)
e.sum(axis=1).shape == (3, 2)
e.sum(axis=2).shape == (3, 2)