Unravel Index
import numpy as np
a = np.array([
[1,2,4,7],
[9,88,6,45],
[9,76,3,4]
])
a
array([[ 1, 2, 4, 7],
[ 9, 88, 6, 45],
[ 9, 76, 3, 4]])
np.unravel_index(np.argmax(a), a.shape)
(1, 1)
The above line will get you the maximum value’s coordinate. In our case 88’s position
np.argmax(a)
5