最大のインデックスを求める

python の 多次元配列 ndarray の要素の最大インデックスを求めます．

プログラム例

以下にプログラム例を示します．006 行目で配列の要素が最大のインデックを求めます．

001   import numpy as np
002   
003   a = [[6,1,1],[2,9,5],[2,9,3]]
004   arr = np.array(a)
005   
006   max_indexes = np.argwhere(arr == arr.max())
007   
008   print(max_indexes)

実行結果

プログラムの実行結果は以下のとおりです．最大の要素の値は 9 で，それは2個あります．実行結果は，その2個のインデックスを示します．

[[1 1]
 [2 1]]