Python 从三维阵列中删除最后一个二维阵列

我有一个带有形状的三维

numpy

数组

(5,5,N)

当我使用

numpy.dstack

将另一个5x5 2D数组添加到此3D数组时，形状会发生如下变化

(5,5,N+1)

等等。我想删除我添加到堆栈中的最后一个2D数组，这样它就可以恢复形状

(5,5,N)

并且可能是

（5,5，N-1），（5,5，N-2），…

，等等

---

实现这一目标的最佳方法是什么？

我将索引如下：

import numpy as np

a = np.ones((5,5,5))
a.shape
(5, 5, 5)

b = np.ones((5, 5, 5))[:, :, :-1]
b.shape
(5, 5, 4)

---

我的索引如下：

import numpy as np

a = np.ones((5,5,5))
a.shape
(5, 5, 5)

b = np.ones((5, 5, 5))[:, :, :-1]
b.shape
(5, 5, 4)