結果 : how to return index of element in array python