結果 : return index of max value in array python