Lambda Function If Else In Python List