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:
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"