結果 : python return if else in one line