总和为另一个特定数字的自然数组合

Question

If i have two natural numbers, "k" and "n". How can i show all combinations in "k" parts that sum to "n"? I tried this code, but i'd like do exclude the sum with zero from the output tuples.

def compositions(k, n):
    if k == 1:
        return [(n,)]

    comp = []
    for i in range(n + 1):
        for t in compositions(k - 1, n - i):
            comp.append((i,) + t)
    return comp

Thank you!

Answer 1

If you want the same outputs but without outputs that include zero, you could do the following.

def compositions(k, n):
    
    if n==0:
        return []
    
    if k == 1:
        return [(n,)]

    comp = []
    for i in range(1,n + 1):
        for t in compositions(k - 1, n - i):
            comp.append((i,) + t)
    return comp