python - numpy 从数组下选择所有元素

Question

我有一个 3 维的 np.array。我只想从最后一个数组中选择所有第 n 个元素。

我知道我可以像这样选择它

array[:, :, 0]

但是我需要使用一个函数，是否有任何 numpy 函数可以做到这一点？

Answer 1

[] 运算符在使用它的对象上调用 __getitem__。在您的情况下，这相当于调用

array.__getitem__((slice(None), slice(None), 0))

这是因为 [:] 是一个空切片(即 slice(None))，而您正在调用 [] 运算符具有 3 个值，这相当于使用 3 个值的元组调用 __getitem__。