If Statement In Lambda Function Python