Lambda Function In Python With If Condition