python - Python中依赖于前一个循环的任意数量的嵌套循环

Question

我试图弄清楚如何迭代任意数量的循环，其中每个循环都取决于最近的外部循环。以下代码是我想要执行的操作的示例:

def function(z):
    n = int(log(z))
    tupes = []
    for i_1 in range(1, n):
        for i_2 in range(1, i_1):
            ...
                ...
                    ...
                        for i_n in range(1, i_{n - 1}):
                            if i_1*i_2*...*i_n > z:
                                tupes.append((i_1, i_2,..., i_n))
    return tupes

虽然我希望它适用于任何 z > e**2，但它足以适用于 z高达 e**100。我知道如果我采用适当的 range 的笛卡尔积，我最终会得到我想要的元组的超集，但我只想获得我寻找的元组。

如果有人能帮我解决这个问题，我将不胜感激。提前致谢。

Answer 1

组合可以按升序排列；事实上，这是 itertools.combinations 的默认行为。

代码:

for i1 in range(1,6):
    for i2 in range(1,i1):
        for i3 in range(1,i2):
            print (i3, i2, i1)

# (1, 2, 3)
# (1, 2, 4)
# ...
# (3, 4, 5)

相当于代码:

from itertools import combinations

for combination in combinations(range(1,6), 3):
    print combination

# (1, 2, 3)
# (1, 2, 4)
# ...
# (3, 4, 5)

使用组合而不是笛卡尔积可以将样本空间缩小到您想要的程度。