python inline if else for loop