Question

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

我知道我可以选择

array[:, :, 0]

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

Answer 1

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

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

这是因为[:]是一个空切片（即slice(None)），并且您正在使用3值调用[]运算符，这等效于使用以下命令调用__getitem__三个值的元组。