python inline if in for loop