python - 在平面上而不是轴上对 3d numpy 数组求和

Question

我有一个 3d numpy 数组 (nx5x5)。我想将 n 个切片中的每一个加起来。所以新的形状将是 (nx1x1)，其中每个 n 只是一个单独的 5x5 数组的总和。我可以在不使用循环的情况下在 numpy 中执行此操作吗？ np.sum 有其轴参数，但它们将数组 reshape 为错误的形状。我想我可能需要调用 np.sum 两次？但是我在思考如何做到这一点时遇到了麻烦。有人知道答案吗？

Answer 1

以下是三种不同的实现方式:

1. 为轴使用元组:

   a.sum(axis=(1, 2))
   

2. 适本地 reshape 以合并您想要求和的轴:

   a.reshape(a.shape[0], -1).sum(axis=1)
   

3. 使用多个和:

   a.sum(-1).sum(-1)
   

   或

   a.sum(1).sum(-1)
   

   等等