For Loop In One Line In Python