Using If Statement In Lambda Python Documentation