Python Inline For If