Syntax Of Lambda Function In Python With If Statement