Using If Statement In Lambda Python Example