Using If Statement In Lambda Python Syntax