Spliting a 3D numpy array on desired axis
Its as simple as doing
A[:,:,0:1].
Thanks @Mark for the answer.
I have a NumPy array of shape (4809, 200, 31) and i want to extract following array out of it : shape (4809, 200, 1). so I want to extract one column from axis=2 from the source array which will have three-axis (0,1,2).