結果 : how to write a one line if statement in python