Writing A For Loop In Python