結果 : if statement in 1 line python