python for loop if else one line