結果 : how to return indices of array in python