結果 : how to write if else in lambda python