np.unravel_index()

import numpy as np

nda1 = np.array([[10,13,12],[13,14,15],[19,17,18]])

print(np.argmax(nda1))

list1 = np.argmax(nda1,axis=1)

# list1 is the list of Max position in each dimension(position in its own dimension)

index2 = np.unravel_index(list1[0],nda1.shape)
index3 = np.unravel_index(list1[1]+3,nda1.shape)
index4 = np.unravel_index(list1[2]+6,nda1.shape)

# index2,index3,index4 is the actual tuple (x,y) position in ndarray

#np.unravel_index is going to convert a INT (single value) to a tuple; +3, +6 is to adjust to the 'real position' in ndarray

 

print(nda1[index2])
print(nda1[index3])
print(nda1[index4])

 

output:

13
15
19