結果 : get index of max value in 2d array python