涉及itertools的两个嵌套for循环不会产生外循环的置换

代码运行不符合预期的原因与此不起作用的原因相同:

p = iter(range(6))
p2 = iter(range(6))

for pos in p:
    for shift in p2:
        print(pos,shift)

itertools.product 返回只能使用一次的迭代器。第一次循环 p2 您已经消耗了所有的值,外部循环的后续迭代本质上变成了禁止操作。

您需要将迭代器输出存储到一个可以多次迭代的容器中,或者调用 itertools。产品 根据需要进行多次重建。

import itertools

a = [0,1,2,3,4]
b = range(6)
p = itertools.permutations(a)
p2 = list(itertools.product(b,b,b,b,b))

for pos in p:
    for shift in p2:
        print(pos,shift)

或

import itertools

a = [0,1,2,3,4]
b = range(6)
p = itertools.permutations(a)

for pos in p:
    p2 = itertools.product(b,b,b,b,b)
    for shift in p2:
        print(pos,shift)

注:实际上 permutations 和 product 方法实际上是用于生成特定于操作的类的构造函数调用,然而,它们确实实现了迭代器接口,所以我只是这样调用它们!