結果 : write if statement in one line python