結果 : how to return in a for loop python