結果 : how to write if statement in lambda python