Shorthand For Loop In Python Using Range