>>> a.argmax(axis=0)
array([1, 1, 0])
NumPy
numpy.org › doc › stable › reference › generated › numpy.argmax.html
numpy.argmax — NumPy v2.5 Manual
Returns the indices of the maximum values along an axis. ... Input array. ... By default, the index is into the flattened array, otherwise along the specified axis.
DataCamp
datacamp.com › doc › numpy › argmax
NumPy argmax()
NumPy's `argmax()` function is used in array computation and analysis to find the index of the maximum value within an array.
03:11
NumPy argmax() Tutorial: Find Index of Maximum Value in Arrays ...
03:11
numpy max index - YouTube
how to find index of max value in numpy array
01:47
How to Index Each Occurrence of a Max Value along a Given Axis ...
03:17
How do I get indices of N maximum values in a NumPy array? - YouTube
00:49
Find the Index of the Biggest and Smallest Element of a #numpy ...
NumPy
numpy.org › doc › 2.2 › reference › generated › numpy.argmax.html
numpy.argmax — NumPy v2.2 Manual
Returns the indices of the maximum values along an axis. ... Input array. ... By default, the index is into the flattened array, otherwise along the specified axis.
ListenData
listendata.com › home › python
NumPy argmax() Function : Learn with Examples
In simple words, it returns the index of the maximum value for each row. # 2D array my_array = np.array([[10,5,17],[2,16,3]]) # Finding the indices of the maximum elements along each row (axis=1) print("Indices of maximum elements per row:", ...
Vultr Docs
docs.vultr.com › python › third party › numpy › argmax()
Python Numpy argmax() - Find Maximum Index
April 10, 2025 - When np.argmax() is applied to a 2D array without an axis, it returns the index of the max value as if the array were flattened.
Analytics Vidhya
analyticsvidhya.com › home › all you need to know about numpy’s argmax() function
All You Need to Know About NumPy's argmax() Function
May 28, 2025 - To find the maximum value index along a specific axis in a multi-dimensional array, we can specify the axis parameter in the np.argmax() function.
GeeksforGeeks
geeksforgeeks.org › python › numpy-argmax-python
numpy.argmax() in Python - GeeksforGeeks
March 8, 2024 - The numpy.argmax() function returns indices of the max element of the array in a particular axis.
NumPy
numpy.org › doc › 2.0 › reference › generated › numpy.argmax.html
numpy.argmax — NumPy v2.0 Manual
Returns the indices of the maximum values along an axis. ... Input array. ... By default, the index is into the flattened array, otherwise along the specified axis.
Educative
educative.io › answers › how-to-get-indices-of-n-maximum-values-in-numpy-arrays
How to get indices of N maximum values in NumPy arrays
In order to get the indices of N maximum values in a NumPy array, we can use the argsort() function.
Real Python
realpython.com › numpy-max-maximum
NumPy's max() and maximum(): Find Extreme Values in Arrays – Real Python
October 22, 2025 - In this introduction to NumPy, you'll learn how to find extreme values using the max() and maximum() functions. This includes finding the maximum element in an array or along a given axis of an array, as well as comparing two arrays to find the larger element in each index position.
Top answer 1 of 16
1032
Newer NumPy versions (1.8 and up) have a function called argpartition for this. To get the indices of the four largest elements, do
>>> a = np.array([9, 4, 4, 3, 3, 9, 0, 4, 6, 0])
>>> a
array([9, 4, 4, 3, 3, 9, 0, 4, 6, 0])
>>> ind = np.argpartition(a, -4)[-4:]
>>> ind
array([1, 5, 8, 0])
>>> top4 = a[ind]
>>> top4
array([4, 9, 6, 9])
Unlike argsort, this function runs in linear time in the worst case, but the returned indices are not sorted, as can be seen from the result of evaluating a[ind]. If you need that too, sort them afterwards:
>>> ind[np.argsort(a[ind])]
array([1, 8, 5, 0])
To get the top-k elements in sorted order in this way takes O(n + k log k) time.
2 of 16
521
The simplest I've been able to come up with is:
>>> import numpy as np
>>> arr = np.array([1, 3, 2, 4, 5])
>>> arr.argsort()[-3:][::-1]
array([4, 3, 1])
This involves a complete sort of the array. I wonder if numpy provides a built-in way to do a partial sort; so far I haven't been able to find one.
If this solution turns out to be too slow (especially for small n), it may be worth looking at coding something up in Cython.
w3resource
w3resource.com › python-exercises › numpy › python-numpy-exercise-120.php
Python NumPy: Get the index of a maximum element in a numpy array along one axis - w3resource
August 29, 2025 - ... # Importing the NumPy library ... indices of the maximum element in the array 'a' # The argmax() function finds the index of the maximum value in the flattened array # The unrave...
Programiz
programiz.com › python-programming › numpy › methods › argmax
NumPy argmax()
import numpy as np array1 = np.array([10, 12, 14, 11, 5]) # return index of largest element (14) maxIndex= np.argmax(array1) print(maxIndex) # Output: 2
NumPy
numpy.org › devdocs › reference › generated › numpy.argmax.html
numpy.argmax — NumPy v2.6.dev0 Manual
Returns the indices of the maximum values along an axis. ... Input array. ... By default, the index is into the flattened array, otherwise along the specified axis.
NumPy
numpy.org › doc › 2.3 › reference › generated › numpy.argmax.html
numpy.argmax — NumPy v2.3 Manual
Returns the indices of the maximum values along an axis. ... Input array. ... By default, the index is into the flattened array, otherwise along the specified axis.
Statology
statology.org › home › how to get the index of max value in numpy array
How to Get the Index of Max Value in NumPy Array
July 29, 2022 - The following code shows how to ... 7, 9, 4, 4, 6, 3]) #find index that contains max value x.argmax() 2 · The argmax() function returns a value of 2....
Codecademy
codecademy.com › docs › python:numpy › ndarray › .argmax()
Python:NumPy | ndarray | .argmax() | Codecademy
October 31, 2025 - Returns the indices of the maximum values along a specified axis.
Geekflare
geekflare.com › development › how to use the numpy argmax() function in python
How to Use the NumPy argmax() Function in Python
December 29, 2024 - The NumPy argmax() function returns the index of the maximum element in an array. If the maximum element occurs more than once in an array a, then np.argmax(a) returns the index of the first occurrence of the element.
Omz Software
omz-software.com › pythonista › numpy › reference › generated › numpy.argmax.html
numpy.argmax — NumPy v1.8 Manual
Indices of the maximum values along an axis.