python - What is the role of axis parameter in np.argmax()?

Question

Wee know, in axis parameter 0,1 means column and row wise maximum element index but for 2,3 & so on what it indicates? An example code is given here. What is the output significance in this code?

Answer 1

When you have an array of higher dimensions you will also have new axes. For example, in a dimension 3 array (e.g. a cube) you will have 3 axes (row, column, depth).

When you pass the axis in the np.argmax you are telling numpy along which axis you want the maximum argument. 3 will throw an error because your array only has 3 axes (0, 1, 2).

Here is an article about numpy arrays axes.