Python结合了两个for循环

Question

目前我会这样做:

for x in [1,2,3]:
    for y in [1,2,3]:
        print x,y

有办法做下面这样的事情,

for x,y in ([1,2,3],[1,2,3]):
    print x,y

想要缩短这种循环,这会引发"太多解压缩"异常.

Answer 1

使用itertools.product

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

打印所有九对:

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

更新:如果要从一个列表中选择两个变量x,y则可以使用repeat关键字(由agf提出):

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

Answer 2

您可以在for循环中使用生成器表达式:

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