Skip to content Skip to sidebar Skip to footer

Python Combine Two For Loops

Currently I would do: for x in [1,2,3]: for y in [1,2,3]: print x,y Is there way of doing something like below, for x,y in ([1,2,3],[1,2,3]): print x,y Would like

Solution 1:

Use itertools.product

import itertools
for x, y in itertools.product([1,2,3], [1,2,3]):
    print x, y

prints all nine pairs:

1 1
1 2
1 3
2 1
2 2
2 3
3 1
3 2
3 3

UPDATE: If the two variables x and y are to be chosen from one list, you can use the repeat keyword (as proposed by agf):

import itertools
for x, y in itertools.product([1,2,3], repeat=2):
    print x, y

Solution 2:

You could use a generator expression in the for loop:

for x, y in ((a,b) for a in [1,2,3] for b in [5,6,7]):
  print x, y

Post a Comment for "Python Combine Two For Loops"