109

How do I enumerate two lists of equal length simultaneously? I am sure there must be a more pythonic way to do the following:

for index, value1 in enumerate(data1):
    print index, value1 + data2[index]

I want to use the index and data1[index] and data2[index] inside the for loop.

6 Answers 6

228

Use zip for both Python2 and Python3:

for index, (value1, value2) in enumerate(zip(data1, data2)):
    print(index, value1 + value2)  # for Python 2 use: `print index, value1 + value2` (no braces)

Note that zip runs only up to the shorter of the two lists(not a problem for equal length lists), but, in case of unequal length lists if you want to traverse the whole list then use itertools.izip_longest.

1
  • 2
    thanks this solved zip(data1, data2).index(val1, val2)
    – Chebhou
    Commented Dec 6, 2015 at 8:48
17
for i, (x, y) in enumerate(zip(data1, data2)):

In Python 2.x, you might want to use itertools.izip instead of zip, esp. for very long lists.

9
  • 19
    Does anyone else hate that enumerate is nine characters long? Commented May 1, 2013 at 21:47
  • 1
    @jamylak: true, so maybe it should be named enu ;)
    – Fred Foo
    Commented May 2, 2013 at 8:54
  • 1
    @larsmans yeah that would fit in with zip and map! should make a request for Python 4 ;) ;)
    – jamylak
    Commented May 2, 2013 at 8:57
  • 2
    @jamylak: don't forget to request rangerng as well to make a stronger case. Oh, and rpr.
    – Fred Foo
    Commented May 2, 2013 at 9:04
  • 5
    @StevenRumbalski: Readability first! - enum() was considered but rejected: PEP 279 Commented May 2, 2013 at 17:36
11
from itertools import count

for index, value1, value2 in zip(count(), data1, data2):
    print(index, value1, value2)

Source: http://www.saltycrane.com/blog/2008/04/how-to-use-pythons-enumerate-and-zip-to/#c2603

2

Althought this is not very clear what you look for,

>>> data1 = [3,4,5,7]
>>> data2 = [4,6,8,9]
>>> for index, value in enumerate(zip(data1, data2)):
    print index, value[0]+value[1]


0 7
1 10
2 13
3 16
1

Since it has been mentioned that the length are equal,

for l in range(0, len(a)):
   print a[l], b[l]
-1

Suppose you want to use zip:

   >>> for x in zip([1,2], [3,4]):
    ...     print x
    ... 
    (1, 3)
    (2, 4)
1
  • 2
    The index is need inside the loop too. Commented May 1, 2013 at 21:42

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