Loop Through An Array In Python With An Index