如何生成一组给定长度和元素总和的所有元组?

How to generate a set of all tuples of given length and sum of elements?

我想要一个函数,它可以生成一组(或列表)所有可能的元组,这些元组具有给定的长度和元素的总和。元组的元素不应该是负整数。

例如下面的输入

get_tuple(length=3, total=2)

我想获得以下输出:

[(1, 0, 1), (2, 0, 0), (1, 1, 0), (0, 0, 2), (0, 1, 1), (0, 2, 0)]

Python 中的标准库是否可以做到这一点?如果没有,怎么写一个函数可以做到?

您可以使用 itertools.product 获取所有数字的乘积,然后对 select 具有指定总和的乘积使用过滤函数:

>>> def get_tuple(length, total):
...     return filter(lambda x:sum(x)==total,product(range(total+1),repeat=length))
... 
>>> get_tuple(3, 2)
[(0, 0, 2), (0, 1, 1), (0, 2, 0), (1, 0, 1), (1, 1, 0), (2, 0, 0)]
>>> get_tuple(4, 12)
[(0, 0, 0, 12), (0, 0, 1, 11), (0, 0, 2, 10), (0, 0, 3, 9), (0, 0, 4, 8), (0, 0, 5, 7), (0, 0, 6, 6), (0, 0, 7, 5), (0, 0, 8, 4), (0, 0, 9, 3), (0, 0, 10, 2), (0, 0, 11, 1), (0, 0, 12, 0), (0, 1, 0, 11), (0, 1, 1, 10), (0, 1, 2, 9), (0, 1, 3, 8), (0, 1, 4, 7), (0, 1, 5, 6), (0, 1, 6, 5), (0, 1, 7, 4), (0, 1, 8, 3), (0, 1, 9, 2), (0, 1, 10, 1), (0, 1, 11, 0), (0, 2, 0, 10), (0, 2, 1, 9), (0, 2, 2, 8), (0, 2, 3, 7), (0, 2, 4, 6), (0, 2, 5, 5), (0, 2, 6, 4), (0, 2, 7, 3), (0, 2, 8, 2), (0, 2, 9, 1), (0, 2, 10, 0), (0, 3, 0, 9), (0, 3, 1, 8), (0, 3, 2, 7), (0, 3, 3, 6), (0, 3, 4, 5), (0, 3, 5, 4), (0, 3, 6, 3), (0, 3, 7, 2), (0, 3, 8, 1), (0, 3, 9, 0), (0, 4, 0, 8), (0, 4, 1, 7), (0, 4, 2, 6), (0, 4, 3, 5), (0, 4, 4, 4), (0, 4, 5, 3), (0, 4, 6, 2), (0, 4, 7, 1), (0, 4, 8, 0), (0, 5, 0, 7), (0, 5, 1, 6), (0, 5, 2, 5), (0, 5, 3, 4), (0, 5, 4, 3), (0, 5, 5, 2), (0, 5, 6, 1), (0, 5, 7, 0), (0, 6, 0, 6), (0, 6, 1, 5), (0, 6, 2, 4), (0, 6, 3, 3), (0, 6, 4, 2), (0, 6, 5, 1), (0, 6, 6, 0), (0, 7, 0, 5), (0, 7, 1, 4), (0, 7, 2, 3), (0, 7, 3, 2), (0, 7, 4, 1), (0, 7, 5, 0), (0, 8, 0, 4), (0, 8, 1, 3), (0, 8, 2, 2), (0, 8, 3, 1), (0, 8, 4, 0), (0, 9, 0, 3), (0, 9, 1, 2), (0, 9, 2, 1), (0, 9, 3, 0), (0, 10, 0, 2), (0, 10, 1, 1), (0, 10, 2, 0), (0, 11, 0, 1), (0, 11, 1, 0), (0, 12, 0, 0), (1, 0, 0, 11), (1, 0, 1, 10), (1, 0, 2, 9), (1, 0, 3, 8), (1, 0, 4, 7), (1, 0, 5, 6), (1, 0, 6, 5), (1, 0, 7, 4), (1, 0, 8, 3), (1, 0, 9, 2), (1, 0, 10, 1), (1, 0, 11, 0), (1, 1, 0, 10), (1, 1, 1, 9), (1, 1, 2, 8), (1, 1, 3, 7), (1, 1, 4, 6), (1, 1, 5, 5), (1, 1, 6, 4), (1, 1, 7, 3), (1, 1, 8, 2), (1, 1, 9, 1), (1, 1, 10, 0), (1, 2, 0, 9), (1, 2, 1, 8), (1, 2, 2, 7), (1, 2, 3, 6), (1, 2, 4, 5), (1, 2, 5, 4), (1, 2, 6, 3), (1, 2, 7, 2), (1, 2, 8, 1), (1, 2, 9, 0), (1, 3, 0, 8), (1, 3, 1, 7), (1, 3, 2, 6), (1, 3, 3, 5), (1, 3, 4, 4), (1, 3, 5, 3), (1, 3, 6, 2), (1, 3, 7, 1), (1, 3, 8, 0), (1, 4, 0, 7), (1, 4, 1, 6), (1, 4, 2, 5), (1, 4, 3, 4), (1, 4, 4, 3), (1, 4, 5, 2), (1, 4, 6, 1), (1, 4, 7, 0), (1, 5, 0, 6), (1, 5, 1, 5), (1, 5, 2, 4), (1, 5, 3, 3), (1, 5, 4, 2), (1, 5, 5, 1), (1, 5, 6, 0), (1, 6, 0, 5), (1, 6, 1, 4), (1, 6, 2, 3), (1, 6, 3, 2), (1, 6, 4, 1), (1, 6, 5, 0), (1, 7, 0, 4), (1, 7, 1, 3), (1, 7, 2, 2), (1, 7, 3, 1), (1, 7, 4, 0), (1, 8, 0, 3), (1, 8, 1, 2), (1, 8, 2, 1), (1, 8, 3, 0), (1, 9, 0, 2), (1, 9, 1, 1), (1, 9, 2, 0), (1, 10, 0, 1), (1, 10, 1, 0), (1, 11, 0, 0), (2, 0, 0, 10), (2, 0, 1, 9), (2, 0, 2, 8), (2, 0, 3, 7), (2, 0, 4, 6), (2, 0, 5, 5), (2, 0, 6, 4), (2, 0, 7, 3), (2, 0, 8, 2), (2, 0, 9, 1), (2, 0, 10, 0), (2, 1, 0, 9), (2, 1, 1, 8), (2, 1, 2, 7), (2, 1, 3, 6), (2, 1, 4, 5), (2, 1, 5, 4), (2, 1, 6, 3), (2, 1, 7, 2), (2, 1, 8, 1), (2, 1, 9, 0), (2, 2, 0, 8), (2, 2, 1, 7), (2, 2, 2, 6), (2, 2, 3, 5), (2, 2, 4, 4), (2, 2, 5, 3), (2, 2, 6, 2), (2, 2, 7, 1), (2, 2, 8, 0), (2, 3, 0, 7), (2, 3, 1, 6), (2, 3, 2, 5), (2, 3, 3, 4), (2, 3, 4, 3), (2, 3, 5, 2), (2, 3, 6, 1), (2, 3, 7, 0), (2, 4, 0, 6), (2, 4, 1, 5), (2, 4, 2, 4), (2, 4, 3, 3), (2, 4, 4, 2), (2, 4, 5, 1), (2, 4, 6, 0), (2, 5, 0, 5), (2, 5, 1, 4), (2, 5, 2, 3), (2, 5, 3, 2), (2, 5, 4, 1), (2, 5, 5, 0), (2, 6, 0, 4), (2, 6, 1, 3), (2, 6, 2, 2), (2, 6, 3, 1), (2, 6, 4, 0), (2, 7, 0, 3), (2, 7, 1, 2), (2, 7, 2, 1), (2, 7, 3, 0), (2, 8, 0, 2), (2, 8, 1, 1), (2, 8, 2, 0), (2, 9, 0, 1), (2, 9, 1, 0), (2, 10, 0, 0), (3, 0, 0, 9), (3, 0, 1, 8), (3, 0, 2, 7), (3, 0, 3, 6), (3, 0, 4, 5), (3, 0, 5, 4), (3, 0, 6, 3), (3, 0, 7, 2), (3, 0, 8, 1), (3, 0, 9, 0), (3, 1, 0, 8), (3, 1, 1, 7), (3, 1, 2, 6), (3, 1, 3, 5), (3, 1, 4, 4), (3, 1, 5, 3), (3, 1, 6, 2), (3, 1, 7, 1), (3, 1, 8, 0), (3, 2, 0, 7), (3, 2, 1, 6), (3, 2, 2, 5), (3, 2, 3, 4), (3, 2, 4, 3), (3, 2, 5, 2), (3, 2, 6, 1), (3, 2, 7, 0), (3, 3, 0, 6), (3, 3, 1, 5), (3, 3, 2, 4), (3, 3, 3, 3), (3, 3, 4, 2), (3, 3, 5, 1), (3, 3, 6, 0), (3, 4, 0, 5), (3, 4, 1, 4), (3, 4, 2, 3), (3, 4, 3, 2), (3, 4, 4, 1), (3, 4, 5, 0), (3, 5, 0, 4), (3, 5, 1, 3), (3, 5, 2, 2), (3, 5, 3, 1), (3, 5, 4, 0), (3, 6, 0, 3), (3, 6, 1, 2), (3, 6, 2, 1), (3, 6, 3, 0), (3, 7, 0, 2), (3, 7, 1, 1), (3, 7, 2, 0), (3, 8, 0, 1), (3, 8, 1, 0), (3, 9, 0, 0), (4, 0, 0, 8), (4, 0, 1, 7), (4, 0, 2, 6), (4, 0, 3, 5), (4, 0, 4, 4), (4, 0, 5, 3), (4, 0, 6, 2), (4, 0, 7, 1), (4, 0, 8, 0), (4, 1, 0, 7), (4, 1, 1, 6), (4, 1, 2, 5), (4, 1, 3, 4), (4, 1, 4, 3), (4, 1, 5, 2), (4, 1, 6, 1), (4, 1, 7, 0), (4, 2, 0, 6), (4, 2, 1, 5), (4, 2, 2, 4), (4, 2, 3, 3), (4, 2, 4, 2), (4, 2, 5, 1), (4, 2, 6, 0), (4, 3, 0, 5), (4, 3, 1, 4), (4, 3, 2, 3), (4, 3, 3, 2), (4, 3, 4, 1), (4, 3, 5, 0), (4, 4, 0, 4), (4, 4, 1, 3), (4, 4, 2, 2), (4, 4, 3, 1), (4, 4, 4, 0), (4, 5, 0, 3), (4, 5, 1, 2), (4, 5, 2, 1), (4, 5, 3, 0), (4, 6, 0, 2), (4, 6, 1, 1), (4, 6, 2, 0), (4, 7, 0, 1), (4, 7, 1, 0), (4, 8, 0, 0), (5, 0, 0, 7), (5, 0, 1, 6), (5, 0, 2, 5), (5, 0, 3, 4), (5, 0, 4, 3), (5, 0, 5, 2), (5, 0, 6, 1), (5, 0, 7, 0), (5, 1, 0, 6), (5, 1, 1, 5), (5, 1, 2, 4), (5, 1, 3, 3), (5, 1, 4, 2), (5, 1, 5, 1), (5, 1, 6, 0), (5, 2, 0, 5), (5, 2, 1, 4), (5, 2, 2, 3), (5, 2, 3, 2), (5, 2, 4, 1), (5, 2, 5, 0), (5, 3, 0, 4), (5, 3, 1, 3), (5, 3, 2, 2), (5, 3, 3, 1), (5, 3, 4, 0), (5, 4, 0, 3), (5, 4, 1, 2), (5, 4, 2, 1), (5, 4, 3, 0), (5, 5, 0, 2), (5, 5, 1, 1), (5, 5, 2, 0), (5, 6, 0, 1), (5, 6, 1, 0), (5, 7, 0, 0), (6, 0, 0, 6), (6, 0, 1, 5), (6, 0, 2, 4), (6, 0, 3, 3), (6, 0, 4, 2), (6, 0, 5, 1), (6, 0, 6, 0), (6, 1, 0, 5), (6, 1, 1, 4), (6, 1, 2, 3), (6, 1, 3, 2), (6, 1, 4, 1), (6, 1, 5, 0), (6, 2, 0, 4), (6, 2, 1, 3), (6, 2, 2, 2), (6, 2, 3, 1), (6, 2, 4, 0), (6, 3, 0, 3), (6, 3, 1, 2), (6, 3, 2, 1), (6, 3, 3, 0), (6, 4, 0, 2), (6, 4, 1, 1), (6, 4, 2, 0), (6, 5, 0, 1), (6, 5, 1, 0), (6, 6, 0, 0), (7, 0, 0, 5), (7, 0, 1, 4), (7, 0, 2, 3), (7, 0, 3, 2), (7, 0, 4, 1), (7, 0, 5, 0), (7, 1, 0, 4), (7, 1, 1, 3), (7, 1, 2, 2), (7, 1, 3, 1), (7, 1, 4, 0), (7, 2, 0, 3), (7, 2, 1, 2), (7, 2, 2, 1), (7, 2, 3, 0), (7, 3, 0, 2), (7, 3, 1, 1), (7, 3, 2, 0), (7, 4, 0, 1), (7, 4, 1, 0), (7, 5, 0, 0), (8, 0, 0, 4), (8, 0, 1, 3), (8, 0, 2, 2), (8, 0, 3, 1), (8, 0, 4, 0), (8, 1, 0, 3), (8, 1, 1, 2), (8, 1, 2, 1), (8, 1, 3, 0), (8, 2, 0, 2), (8, 2, 1, 1), (8, 2, 2, 0), (8, 3, 0, 1), (8, 3, 1, 0), (8, 4, 0, 0), (9, 0, 0, 3), (9, 0, 1, 2), (9, 0, 2, 1), (9, 0, 3, 0), (9, 1, 0, 2), (9, 1, 1, 1), (9, 1, 2, 0), (9, 2, 0, 1), (9, 2, 1, 0), (9, 3, 0, 0), (10, 0, 0, 2), (10, 0, 1, 1), (10, 0, 2, 0), (10, 1, 0, 1), (10, 1, 1, 0), (10, 2, 0, 0), (11, 0, 0, 1), (11, 0, 1, 0), (11, 1, 0, 0), (12, 0, 0, 0)]
>>> 

您可以创建一个递归函数,避免创建所有候选组合并检查它们的总和:

def get_tuples(length, total):
    if length == 1:
        yield (total,)
        return

    for i in xrange(total + 1):
        for t in get_tuples(length - 1, total - i):
            yield (i,) + t

如果我们测试:

>>> list(get_tuples(3, 2))
[(0, 0, 2), (0, 1, 1), (0, 2, 0), (1, 0, 1), (1, 1, 0), (2, 0, 0)]