python - 按变量的嵌套循环数

Question

这个问题已经有答案了:

Function with varying number of For Loops (python) [duplicate] (8 个回答)

How can I make a for-loop pyramid more concise in Python? [duplicate] (4 个回答)

已关闭 6 年前。

我希望我的程序找到 x 个整数的组合，其值在一定范围内，总和等于给定数字。

例如:我想找到 3 个整数的组合，其值可以在 0 到 2 之间，并且总和等于 5。我可以通过编码来做到这一点:

possibilities = []
total = 5
valueRange = 3
for num1 in xrange(valueRange):
    for num2 in xrange(valueRange):
        for num3 in xrange(valueRange):
            if num1 + num2 + num3 == total:
                possibilities.append([num1, num2, num3])

我可以通过更改我创建的变量来更改总和值和范围值，但是如何使用变量指定嵌套循环的数量？谢谢

Answer 1

itertools.product()函数应该有帮助:

>>> [values for values in product(range(3), repeat=3) if sum(values) == 5]
[(1, 2, 2), (2, 1, 2), (2, 2, 1)]

看到答案包含相同答案的字谜，您可以使用 itertools.combinations_with_replacement() 进一步减少工作量。 :

>>> # 4 digits in [0, 1, 2, 3, 4] summing to 6
>>> for values in combinations_with_replacement(range(5), 4):
        if sum(values) == 6:
            print(values)

(0, 0, 2, 4)
(0, 0, 3, 3)
(0, 1, 1, 4)
(0, 1, 2, 3)
(0, 2, 2, 2)
(1, 1, 1, 3)
(1, 1, 2, 2)