if in inline for python