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