/Python/Get index of max value python numpy

Get index of max value python numpy

Simple
>> b
array([0, 5, 2, 3, 4, 5])

>> np.argmax(b)  # Only the first occurrence is returned.
1
-----------------------------
#More complicated
>> a
array([[10, 11, 12],
       [13, 14, 15]])

>> np.argmax(a)
5

>> np.argmax(a, axis=0)
array([1, 1, 1])

>> np.argmax(a, axis=1)
array([2, 2])

Similar codes

3 votes187 views1 answer
Factorial program
Python
4 votes210 views6 answers
Online python converter
Python
0 vote212 views2 answers
Write a program to check whether a character is vowel or consonant in python
Python
0 vote199 views1 answer
Django filter queryset by date
Python
0 vote211 views1 answer
Pandas_datareader
Python
-4 votes211 views4 answers
Python 3.4 release date
Python
0 vote201 views1 answer
How to print numpy array in python
Python
6 votes194 views3 answers
%s in python
Python
3 votes153 views2 answers
Django favicon
Python Django
25 votes158 views7 answers
Remove space from start and end of string in python
Python