結果 : write if in one line python