結果 : how to get index of number in array python