Python Lambda Function Multiple Columns