indexing current value in for loop? Python

Question

So I have a for loop.

>>> row = [5, 2, 5, 4, 2, 2, 5, 5, 5, 2]
>>> for i in row:
    if i == 5:
        print(row.index(i))
    if i == 2:
        print(row.index(i))

OUTPUT

I want to get: 0 1 2 4 5 6 7 8 9. In other words, I want to get the index of the i I'm currently looking at in the for loop, if it is = 5 or 2... any easy way to do this?

Martijn Pieters · Accepted Answer · 2014-03-08 09:23:33Z

7

Use the enumerate() function to produce a running index:

for index, i in enumerate(row):
    if i == 5:
        print(index)
    if i == 2:
        print(index)

or simpler still:

for index, i in enumerate(row):
    if i in (2, 5):
        print(index)

answered Mar 8, 2014 at 9:23

Martijn Pieters

1.1m326 gold badges4.2k silver badges3.4k bronze badges

Sign up to request clarification or add additional context in comments.

1 Comment

thefourtheye Over a year ago

Ding ding ding, we have a winner in the 33rd second :)

Jon Clements · Accepted Answer · 2014-03-08 09:41:57Z

1

For completeness sake, an alternative if one didn't want to use enumerate for some reason:

row = [5, 2, 5, 4, 2, 2, 5, 5, 5, 2]
indices = filter(lambda idx: row[idx] in (5,2), xrange(len(row)))
# [0, 1, 2, 4, 5, 6, 7, 8, 9]

Or as a list-comp:

indices = [idx for idx in xrange(len(row)) if row[idx] in (5, 2)]

edited Mar 8, 2014 at 9:41

answered Mar 8, 2014 at 9:35

Jon Clements

143k34 gold badges254 silver badges288 bronze badges

Collectives™ on Stack Overflow

indexing current value in for loop? Python

2 Answers 2

1 Comment

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

1 Comment

Comments

Your Answer

Sign up or log in

Post as a guest

Related