/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

0 vote204 views1 answer
Returns all column names as a list
Python
0 vote192 views1 answer
How to order randomly in django orm
Python
13 votes211 views3 answers
How to split text into list python by characters
Python
4 votes248 views4 answers
Python find factors of a number
Python
2 votes237 views1 answer
Flask session timeout
Python Flask
1 vote205 views1 answer
Openpyxl write in cell
Python
2 votes190 views4 answers
Urllib.request.urlretrieve
Python
0 vote185 views1 answer
How to solve add decimals in strings as integers
Python
0 vote201 views1 answer
Pandas pull value from column
Python
6 votes247 views3 answers
Python print user input
Python