結果 : how to write single line if statement in python