結果 : how to return an if statement in python