在所需的轴上分割3D numpy数组

Question

我有一个形状(4809,200,31)的NumPy数组，我想从中提取以下数组:形状(4809,200,1)。所以我想从轴=2中提取一列，从源数组中提取三轴(0,1,2)。

Answer 1

这很简单

(:,:, 0:1)。谢谢@Mark的回答。