# numpy.argmax() in Python

The numpy.argmax() function returns the indices of the maximum values along an axis. In case of multiple occurrences of the maximum values, the indices corresponding to the first occurrence will be returned.

## Syntax

`numpy.argmax(a, axis=None, out=None)`

## Parameters

• array: Input array
• axis [int, optional]By default, the index is into the flattened array, otherwise along the specified axis.
• out [array optional]: If provided, the result will be inserted into this array. It should be of the appropriate shape and dtype.

## Return Value

An array of indices into the array. It will have the same shape as the array.shape with the dimension along the axis removed.

## Finding the maximum element from a matrix with Python numpy.argmax()

``````import numpy as np

a = np.matrix([[1,2,3,33],[4,5,6,66],[7,8,9,99]])

print(np.argmax(a))  # 11, which is the position of 99
print(np.argmax(a[:,:]))  # 11, which is the position of 99
print(np.argmax(a[:1]))  # 3, which is the position of 33
print(np.argmax(a[:,2]))  # 2, which is the position of 9
print(np.argmax(a[1:,2]))  # 1, which is the position of 9``````

Output

``````11
11
3
2
1``````

The argmax() returns the position or index of the largest value in an array. The array can be of a single or multidimensional,

### Using np.unravel_index on argmax output

We can use the `np.unravel_index` function for getting an index corresponding to a 2D array from the `numpy.argmax` output.

``````import numpy as np
a = np.arange(6).reshape(2,3) + 10
print(a)

index = np.unravel_index(np.argmax(a), a.shape)
print(index)
print(a[index])``````

Output

``````[[10 11 12]
[13 14 15]]
(1, 2)
15``````

## Finding Maximum Elements along columns using Python numpy.argmax()

The below code returns the index value of the maximum elements along each column.

``````import numpy as np
a = np.arange(12).reshape(4,3) + 10
print(a)

print("Max elements", np.argmax(a, axis=0))
``````

Output

``````[[10 11 12]
[13 14 15]
[16 17 18]
[19 20 21]]
Max elements [3 3 3]``````

Subscribe to get notified of the latest articles. We will never spam you. Be a part of our ever-growing community.

##### You May Also Like ## How to Create a Directory in Python?

Table of Contents Hide Method 1 – Using os.mkdir() functionExample 1 – Create a directory in Python using os.mkdir()Example 2 – Exception if the directory already existsMethod 2 – Using… ## numpy.average() Function

Table of Contents Hide SyntaxParametersReturn ValueRaisesExample 1: NumPy average() of Array values Example 2: NumPy average() of an array values in column-wise Example 3: NumPy average() of an array values in row-wiseExample… ## Python Reverse a List: A Step-by-Step Tutorial

Reversing a list is a common requirement in any programming language. In this tutorial, we will learn the effective way to reverse a list in Python.  There are 3 ways… ## Python abs()

Table of Contents Hide abs() Syntax abs() Parametersabs() Return ValueWhat does the abs() function do in Python?Example 1: Get absolute value of a number in PythonExample 2: Get the magnitude of… ## Menu Driven Program in Python

Introduction to Menu-Driven Program A menu driven program in Python is a program that takes input from a user by displaying a list of options and allows users to choose… ## Python String count()

Table of Contents Hide SyntaxParameterReturn ValueExample 1: Count the number of occurrences of a given substringExample 2: Count the number of occurrences of a given substring with start and end…