結果 : get index of max array python