title: Maximum values along axis of Numpy ndarray?
body: <p><a href="https://i.stack.imgur.com/7y7tM.jpg" rel="nofollow noreferrer"><img src="https://i.stack.imgur.com/7y7tM.jpg" alt="numpy_array" /></a></p>
  
  <p>I'm afraid that I can't describe the problem so I draw a sketch of it.Anyway,what I need is to find the max values along the 0th axis in a numpy ndarray,i.e.array.shape(5,5,3), and their corresponding &quot;layer numbers&quot;, and use the &quot;layer numbers&quot; to create a new 2d array with shape of (1,5,3).Hope I'm giving a clear description here..thanks a lot.</p>