[解決済み] 多次元NumPy配列の中で最大の値の位置を取得する。

質問

多次元NumPy配列の中で最大の値の位置（インデックス）を取得するにはどうすればよいですか？

どのように解決するのですか？

この argmax() メソッドが役立つはずです。

更新

(コメントを読んで)私は argmax() メソッドは、多次元配列でも機能すると思います。リンク先のドキュメントにその例があります。

>>> a = array([[10,50,30],[60,20,40]])
>>> maxindex = a.argmax()
>>> maxindex
3

アップデート2

(感謝 ケニーTM のコメント) unravel_index(a.argmax(), a.shape) を使ってタプルとしてインデックスを取得することができます。

>>> from numpy import unravel_index
>>> unravel_index(a.argmax(), a.shape)
(1, 0)