If Statement In Lambda Python