結果 : use if statement in lambda python