Lambda Function Python With If