Current iteration number in Python for loop

Question

I am sure there is a better way or writing this python function:

end_points = list(range(100))
filepath = 'something.csv'
with open(filepath) as fp:
    cnt = 0
    for line in fp:
        end_points[cnt]=[x.strip() for x in line.split(',')]
        cnt += 1

It works but it is not elegant. Is there a way of automatically refers to the current number of iteration in the for loop?

score 4 · Accepted Answer · edited Dec 20 '19 at 08:28

4

You can also use the enumerate function:

for iteration_no, line in enumerate(fp):
    print(iteration_no)
    end_points[iteration_no]=[x.strip() for x in line.split(',')]

If you are more generally interested in seeing the progress of your loop you should alternatively have a look at the TQDM package which dynamically prints the progress of your loop.

https://github.com/tqdm/tqdm

edited Dec 20 '19 at 08:28

Matthias

12,873
6
42
48

answered Dec 17 '19 at 12:09

Dominique Paul

1,623
2
18
31

score 1 · Answer 2 · answered Dec 17 '19 at 12:00

1

Not sure of being elegant but is shorter,

>>> with open("something.csv") as f:
...     result = [list(map(str.strip,x.split(','))) for x in f]
...     print(result)

answered Dec 17 '19 at 12:00

abhilb

5,639
2
20
26

Current iteration number in Python for loop

2 Answers2