How To Define A Lambda Function Python