結果 : write if condition in one line python