从3D阵列中删除最后的2D阵列

Question

我有一个形状为

的三维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),...等。

实现这一目标的最蟒蛇方式是什么？

Answer 1

我的索引如下：

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)