結果 : write if and else in one line python