Python中的组合函数[重复]

Question

我想写一个函数，它给我一个数字组合（它们的给定数量）作为系数，它们的总和给我们 4。

假设给定的变量数是 2。所以我们需要说找到a+b=4 的组合。它将是[1,3], [2,2], [3,1], [0,4], [4,0]。或者如果给定的变量号是3，那么需要找到a+b+c=4的组合，这将是[1,2,1], [1,1,2]等。

我该怎么做？

Answer 1

您尝试解决的问题在数论中称为Integer Partitioning。有关可能的解决方案，请参阅this answer。 This数学交流也可以帮助你更多地了解问题背后的数学。

Answer 2

您可以使用此代码

def combinations(iterable, r):
# combinations('ABCD', 2) --> AB AC AD BC BD CD
# combinations(range(4), 3) --> 012 013 023 123
pool = tuple(iterable)
n = len(pool)
if r > n:
    return
indices = list(range(r))
yield tuple(pool[i] for i in indices)
while True:
    for i in reversed(range(r)):
        if indices[i] != i + n - r:
            break
    else:
        return
    indices[i] += 1
    for j in range(i+1, r):
        indices[j] = indices[j-1] + 1
    yield tuple(pool[i] for i in indices)