/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 votes167 views1 answer
How to stop python for certain time in python
Python
0 vote160 views1 answer
Np.conjugate
Python
6 votes199 views2 answers
Python generate dictionary in loop
Python
3 votes204 views1 answer
Numpy fill na with 0
Python
68 votes173 views6 answers
Python dict sort by value
Python
0 vote206 views1 answer
Bootstrap 5 in Django
Python Django
17 votes167 views2 answers
How to record execution time in python
Python
0 vote176 views1 answer
Sklearn euclidean distance
Python
1 vote134 views1 answer
Pandas series contains
Python
0 vote177 views1 answer
Positive counter python
Python