Numpy 2D或3D阵列轴明智和

问题描述

我正在尝试使用numpy的np.sum（），但轴值使我感到困惑，我不确定如何考虑3D数组总和的总和/直觉。

当使用2D数组时，我很好。

任何人都可以引导我建立一个好的链接，或者让我了解如何处理吗？

解决方法

经过一些发现，我试图创建一个手绘图来解释轴向计算对于Numpy意味着什么。 为了简单起见，我使用了不同的示例进行解释：

,

我认为，但我不知道，这里的想法是您只将数组的维数减少了一个。因此，如果您有3d数组，那么根据您选择的轴，您的总和将在X，Y或Z方向上为2d数组。

似乎还可以通过将多个轴传递来将尺寸降低2倍，而不是1倍。因此，您可以通过传递axis=[0,1]或axis=[1,2]或axis=[0,2]

从3d数组中获得1d数组

https://numpy.org/doc/stable/reference/generated/numpy.sum.html