Using If In Lambda Function Python