結果 : get index of max in array python