結果 : get index of max value of array python