結果 : index of the max value in an array python