xxxxxxxxxx
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])