What Is The Role Of Axis Parameter In Np.argmax()?

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 signifi

Solution 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.