numpy.argmin() in Python

Read

The numpy.argmin() method returns indices of the min element of the array in a particular axis.

Syntax :

numpy.argmin(array, axis = None, out = None)

Parameters :

array : Input array to work on 
axis  : [int, optional]Along a specified axis like 0 or 1
out   : [array optional]Provides a feature to insert output to the out
          array and it should be of appropriate shape and dtype

Return :

Array of indices into the array with same shape as array.shape
 with the dimension along axis removed.

Code 1 :

Python

<div id="highlighter_944501" class="syntaxhighlighter nogutter  "><table border="0" cellpadding="0" cellspacing="0"><tbody><tr><td class="code"><div class="container"><div class="line number1 index0 alt2"><code class="comments"># Python Program illustrating </code></div><div class="line number2 index1 alt1"><code class="comments"># working of argmin() </code></div><div class="line number3 index2 alt2"><code class="undefined spaces"> </code> </div><div class="line number4 index3 alt1"><code class="keyword">import</code> <code class="plain">numpy as geek  </code></div><div class="line number5 index4 alt2"><code class="undefined spaces"> </code> </div><div class="line number6 index5 alt1"><code class="comments"># Working on 1D array </code></div><div class="line number7 index6 alt2"><code class="plain">array </code><code class="keyword">=</code> <code class="plain">geek.arange(</code><code class="value">8</code><code class="plain">) </code></div><div class="line number8 index7 alt1"><code class="keyword">print</code><code class="plain">(</code><code class="string">"INPUT ARRAY : \n"</code><code class="plain">, array) </code></div><div class="line number9 index8 alt2"><code class="undefined spaces"> </code> </div><div class="line number10 index9 alt1"><code class="undefined spaces"> </code> </div><div class="line number11 index10 alt2"><code class="comments"># returning Indices of the min element </code></div><div class="line number12 index11 alt1"><code class="comments"># as per the indices </code></div><div class="line number13 index12 alt2"><code class="keyword">print</code><code class="plain">(</code><code class="string">"\nIndices of min element : "</code><code class="plain">, geek.argmin(array, axis</code><code class="keyword">=</code><code class="value">0</code><code class="plain">)) </code></div></div></td></tr></tbody></table></div>

Output :

INPUT ARRAY : 
 [0 1 2 3 4 5 6 7]

Indices of min element :  0

Code 2 :

Python

# Python Program illustrating 
# working of argmin() 
  
import numpy as geek  
  
# Working on 2D array 
array =  geek.random.randint(16, size=(4, 4)) 
print("INPUT ARRAY : \n", array) 
  
# returning Indices of the min element 
# as per the indices 
  
'''    
   [[ 8 13  5  0] 
   [ 0  2  5  3] 
   [10  7 15 15] 
   [ 3 11  4 12]] 
     ^  ^  ^  ^ 
     0  2  4  0  - element 
     1  1  3  0  - indices 
'''
print("\nIndices of min element : ", geek.argmin(array, axis = 0)) 

Output :

INPUT ARRAY : 
 [[ 8 13  5  0]
 [ 0  2  5  3]
 [10  7 15 15]
 [ 3 11  4 12]]

Indices of min element :  [1 1 3 0]

Code 3 :

Python

# Python Program illustrating 
# working of argmin() 
  
import numpy as geek  
  
# Working on 2D array 
array =  geek.arange(10).reshape(2, 5) 
print("array : \n", array) 
  
array[0][0] = 10
array[1][1] = 1
array[0][1] = 1
print("\narray : \n", array) 
       
# Returns min element 
print("\narray : ", geek.argmin(array)) 
  
# First occurrence of an min element is given 
print("\nmin ELEMENT INDICES : ", geek.argmin(array, axis = 0)) 

Output :

array : 
 [[0 1 2 3 4]
 [5 6 7 8 9]]

array : 
 [[10  1  2  3  4]
 [ 5  1  7  8  9]]

array :  1

min ELEMENT INDICES :  [1 0 0 0 0]

References :
https://docs.scipy.org/doc/numpy-dev/reference/generated/numpy.argmin.html#numpy.argmin
Note :
These codes won’t run on online IDE’s. Please run them on your systems to explore the working
.

Don't miss your chance to ride the wave of the data revolution! Every industry is scaling new heights by tapping into the power of data. Sharpen your skills, become a part of the hottest trend in the 21st century.
Dive into the future of technology - explore the Complete Machine Learning and Data Science Program by GeeksforGeeks and stay ahead of the curve.

Commit to GfG's Three-90 Challenge! Purchase a course, complete 90% in 90 days, and save 90% cost click here to explore.

Last Updated : 09 Mar, 2022