python lambda for loop if else