Define Lambda Function Python