Write Lambda Function Python