access for loop index in Python