Lambda Function Python If