python single line for loop with if else

my head sounds like a watermelon × Fale Conosco!