Numpy Argmax : How to get argmax element in numpy array stepwise ?

Numpy Argmax implementation

Numpy argmax function returns the indices of the maximum element of NumPy array axis wise. If the axis is None, It gives indices of max in the array. Well, This article will introduce the NumPy argmax with syntax and Implementation.

NumPy argmax : How to use it?

In this section firstly, we will implement the argmax() function. But before this, We have a prerequisite of numpy array.

Prerequisites :

Let’s create a NumPy array with some elements. Here is the piece of the code.

import numpy as np
arr = np.array([[1, 12, 9], [41, 15, 23],[43, 55, 98]])

1. argmax with axis=None :

If we provide the axis =None or ignore the axis parameter. We will get the indices of the max element in NumPy array. Let’s invoke this function.

import numpy as np
arr = np.array([[1, 12, 9], [41, 15, 23],[43, 55, 98]])
np.argmax(arr)

We can also use add axis=None like below. It will not impact anywhere.

np.argmax(arr,axis=None)
argmax with axis=None
argmax with axis=None

Here we have the max element at the 8th indices of the NumPy array. Which is the output for the function.

2. Argmax with axis=0 :

Here axis =0 means, It will show the max element column-wise. Let’s implement it.

import numpy as np
arr = np.array([[1, 12, 9], [41, 15, 23],[43, 55, 98]])
np.argmax(arr,axis=0)
argmax with axis=0
argmax with axis=0

Here in each column, the max element is at indices 2. That’s why it returns [2,2,2]. Because the element 41, 55,98 are the max element in each column.

3.Argmax with axis=1 :

Here everything else will be the same only it will return the max element indices row-wise.

import numpy as np
arr = np.array([[1, 12, 9], [41, 15, 23],[43, 55, 98]])
np.argmax(arr,axis=1)
argmax with axis=1
argmax with axis=1

 

As in the above example, the row-wise (12,41,98) elements are maximum respectively. These elements are found are at [1,0,2] indices. Hence the above function returns the same output for the above code.

Conclusion –

Numpy is really essential in the data processing. NumPy argmax is an implicit function of NumPy python module. As we have seen this function argmax returns the max element axis wise. I hope this article is enough for knowing argmax function. Still, if you think, anything should be the part of this article. Please let us know.

 

Thanks 

Data Science Learner Team

Join our list

Subscribe to our mailing list and get interesting stuff and updates to your email inbox.

Thank you for signup. A Confirmation Email has been sent to your Email Address.

Something went wrong.

Meet Abhishek ( Chief Editor) , a data scientist with major expertise in NLP and Text Analytics. He has worked on various projects involving text data and have been able to achieve great results. He is currently manages Datasciencelearner.com, where he and his team share knowledge and help others learn more about data science.
 
Thank you For sharing.We appreciate your support. Don't Forget to LIKE and FOLLOW our SITE to keep UPDATED with Data Science Learner