結果 : python return if statement one line