結果 : python write if statement in one line