For loop with two index python

In C language we can use two index variable in a single for loop like below.

for (i = 0, j = 0; i < i_max && j < j_max; i++, j++)

Can some one tell how to do this in Python ?

asked Aug 8, 2018 at 13:28

rashokrashok

12k13 gold badges85 silver badges97 bronze badges

3

With zip we can achieve this.

>>> i_max = 5
>>> j_max = 7
>>> for i, j in zip(range(0, i_max), range(0, j_max)):
...     print str(i) + ", " + str(j)
...
0, 0
1, 1
2, 2
3, 3
4, 4

answered Aug 8, 2018 at 13:36

rashokrashok

12k13 gold badges85 silver badges97 bronze badges

2

If the first answer does not suffice; to account for lists of different sizes, a possible option would be:

a = list(range(5))
b = list(range(15))
for i,j in zip(a+[None]*(len(b)-len(a)),b+[None]*(len(a)-len(b))):
    print(i,j)

Or if one wants to cycle around shorter list:

from itertools import cycle

for i,j in zip(range(5),cycle(range(2)):
    print(i,j)

answered Aug 8, 2018 at 13:43

sdgaw erzswersdgaw erzswer

1,9681 gold badge23 silver badges45 bronze badges

You can do this in Python by using the syntax for i,j in iterable:. Of course in this case the iterable must return a pair of values. So for your example, you have to:

• define the list of values for i and for j
• build the iterable returning a pair of values from both lists: zip() is your friend in this case (if the lists have different sizes, it stops at the last element of the shortest one)
• use the syntax for i,j in iterable:

Here is an example:

i_max = 7
j_max = 9

i_values = range(i_max)
j_values = range(j_max)

for i,j in zip(i_values,j_values):
    print(i,j)
    # 0 0
    # 1 1
    # 2 2
    # 3 3
    # 4 4
    # 5 5
    # 6 6

answered Aug 8, 2018 at 13:42

Laurent H.Laurent H.

5,9761 gold badge17 silver badges37 bronze badges

1

One possible way to do that is to iterate over a comprehensive list of lists. For example, if you want to obtain something like

for r in range(1, 3):
    for c in range(5, 7):
        print(r, c)

which produces

# 1 5
# 1 6
# 2 5
# 2 6

is by using

for i, j in [[_i, _j] for _i in range(1, 3) for _j in range(5, 7)]:
    print(i, j)

Maybe, sometimes one more line is not so bad.

answered Sep 9, 2021 at 13:52

Not the answer you're looking for? Browse other questions tagged python for-loop or ask your own question.

If you’re moving to Python from C or Java, you might be confused by Python’s for loops. Python doesn’t actually have for loops… at least not the same kind of for loop that C-based languages have. Python’s for loops are actually foreach loops.

In this article I’ll compare Python’s for loops to those of other languages and discuss the usual ways we solve common problems with for loops in Python.

For loops in other languages

Before we look at Python’s loops, let’s take a look at a for loop in JavaScript:

1
2
3
4

var colors = ["red", "green", "blue", "purple"];
for (var i = 0; i < colors.length; i++) {
    console.log(colors[i]);
}

This JavaScript loop looks nearly identical in C/C++ and Java.

In this loop we:

1. Set a counter variable i to 0
2. Check if the counter is less than the array length
3. Execute the code in the loop or exit the loop if the counter is too high
4. Increment the counter variable by 1

Looping in Python

Now let’s talk about loops in Python. First we’ll look at two slightly more familiar looping methods and then we’ll look at the idiomatic way to loop in Python.

while

If we wanted to mimic the behavior of our traditional C-style for loop in Python, we could use a while loop:

1
2
3
4
5

colors = ["red", "green", "blue", "purple"]
i = 0
while i < len(colors):
    print(colors[i])
    i += 1

This involves the same 4 steps as the for loops in other languages (note that we’re setting, checking, and incrementing i) but it’s not quite as compact.

This method of looping in Python is very uncommon.

range of length

I often see new Python programmers attempt to recreate traditional for loops in a slightly more creative fashion in Python:

1
2
3

colors = ["red", "green", "blue", "purple"]
for i in range(len(colors)):
    print(colors[i])

This first creates a range corresponding to the indexes in our list (0 to len(colors) - 1). We can loop over this range using Python’s for-in loop (really a foreach).

This provides us with the index of each item in our colors list, which is the same way that C-style for loops work. To get the actual color, we use colors[i].

for-in: the usual way

Both the while loop and range-of-len methods rely on looping over indexes. But we don’t actually care about the indexes: we’re only using these indexes for the purpose of retrieving elements from our list.

Because we don’t actually care about the indexes in our loop, there is a much simpler method of looping we can use:

1
2
3

colors = ["red", "green", "blue", "purple"]
for color in colors:
    print(color)

So instead of retrieving the item indexes and looking up each element, we can just loop over our list using a plain for-in loop.

The other two methods we discussed are sometimes referred to as anti-patterns because they are programming patterns which are widely considered unidiomatic.

What if we need indexes?

What if we actually need the indexes? For example, let’s say we’re printing out president names along with their numbers (based on list indexes).

range of length

We could use range(len(our_list)) and then lookup the index like before:

1
2
3

presidents = ["Washington", "Adams", "Jefferson", "Madison", "Monroe", "Adams", "Jackson"]
for i in range(len(presidents)):
    print("President {}: {}".format(i + 1, presidents[i]))

But there’s a more idiomatic way to accomplish this task: use the enumerate function.

enumerate

Python’s built-in enumerate function allows us to loop over a list and retrieve both the index and the value of each item in the list:

1
2
3

presidents = ["Washington", "Adams", "Jefferson", "Madison", "Monroe", "Adams", "Jackson"]
for num, name in enumerate(presidents, start=1):
    print("President {}: {}".format(num, name))

The enumerate function gives us an iterable where each element is a tuple that contains the index of the item and the original item value.

This function is meant for solving the task of:

1. Accessing each item in a list (or another iterable)
2. Also getting the index of each item accessed

So whenever we need item indexes while looping, we should think of enumerate.

Note: the start=1 option to enumerate here is optional. If we didn’t specify this, we’d start counting at 0 by default.

What if we need to loop over multiple things?

Often when we use list indexes, it’s to look something up in another list.

enumerate

For example, here we’re looping over two lists at the same time using indexes to look up corresponding elements:

1
2
3
4
5

colors = ["red", "green", "blue", "purple"]
ratios = [0.2, 0.3, 0.1, 0.4]
for i, color in enumerate(colors):
    ratio = ratios[i]
    print("{}% {}".format(ratio * 100, color))

Note that we only need the index in this scenario because we’re using it to lookup elements at the same index in our second list. What we really want is to loop over two lists simultaneously: the indexes just provide a means to do that.

zip

We don’t actually care about the index when looping here. Our real goal is to loop over two lists at once. This need is common enough that there’s a special built-in function just for this.

Python’s zip function allows us to loop over multiple lists at the same time:

1
2
3
4

colors = ["red", "green", "blue", "purple"]
ratios = [0.2, 0.3, 0.1, 0.4]
for color, ratio in zip(colors, ratios):
    print("{}% {}".format(ratio * 100, color))

The zip function takes multiple lists and returns an iterable that provides a tuple of the corresponding elements of each list as we loop over it.

Note that zip with different size lists will stop after the shortest list runs out of items. You may want to look into itertools.zip_longest if you need different behavior. Also note that zip in Python 2 returns a list but zip in Python 3 returns a lazy iterable. In Python 2, itertools.izip is equivalent to the newer Python 3 zip function.

Looping cheat sheet

Here’s a very short looping cheat sheet that might help you remember the preferred construct for each of these three looping scenarios.

Loop over a single list with a regular for-in:

1
2

for n in numbers:
    print(n)

Loop over multiple lists at the same time with zip:

1
2

for header, rows in zip(headers, columns):
    print("{}: {}".format(header, ", ".join(rows)))

Loop over a list while keeping track of indexes with enumerate:

1
2

for num, line in enumerate(lines):
    print("{0:03d}: {}".format(num, line))

In Summary

If you find yourself tempted to use range(len(my_list)) or a loop counter, think about whether you can reframe your problem to allow usage of zip or enumerate (or a combination of the two).

In fact, if you find yourself reaching for enumerate, think about whether you actually need indexes at all. It’s quite rare to need indexes in Python.

1. If you need to loop over multiple lists at the same time, use zip
2. If you only need to loop over a single list just use a for-in loop
3. If you need to loop over a list and you need item indexes, use enumerate

If you find yourself struggling to figure out the best way to loop, try using the cheat sheet above.

Practice makes perfect

You don’t learn by putting information in your head, you learn by attempting to retrieve information from your head. So you’ve just read an article on something new, but you haven’t learned yet.

Write some code that uses enumerate and zip later today and then quiz yourself tomorrow on the different ways of looping in Python. You have to practice these skills if you want to actually remember them.

If you’d like to get hands-on experience practicing Python every week, I have a Python skill-building service you should consider joining. If you sign up for Python Morsels I’ll give you a Python looping exercise that right now and then I’ll send you one new Python exercise every week after that.

Fill out the form above to sign up for Python Morsels, get some practice with the zip function, and start leveling-up your Python skills every week.

Can you have 2 variables in a for loop Python?

Can you have two arguments in a for loop?

How do you iterate through an index in Python?

How do you pass two arguments in a for loop in Python?