if else in inline for loop python