Comment obtenir l'index d'un élément maximum dans un tableau NumPy le long d'un axe

Question

J'ai un tableau NumPy bidimensionnel. Je sais comment obtenir les valeurs maximales sur les axes :

>>> a = array([[1,2,3],[4,3,1]])
>>> amax(a,axis=0)
array([4, 3, 3])

Comment puis-je obtenir les indices des éléments maximum ? Je voudrais plutôt comme résultat array([1,1,0]).

Answer 1

>>> a.argmax(axis=0)

array([1, 1, 0])

Answer 2

>>> import numpy as np
>>> a = np.array([[1,2,3],[4,3,1]])
>>> i,j = np.unravel_index(a.argmax(), a.shape)
>>> a[i,j]
4

Answer 3

v = alli.max()
index = alli.argmax()
x, y = index/8, index%8