結果 : how to return a for loop in python