исчерпывающая комбинация питона

Question

Я пытаюсь написать функцию с входными данными:

given_num=6, n=3

Как вернуть результат всех возможных комбинаций в виде списка с n элементами и суммой = Given_num, вложенными в список:

[[0,0,6],[0,1,5],[0,2,4],[0,3,3],[0,4,2],[0,5,1],[0,6,0],
[1,0,5],[1,1,4],[1,2,3],[1,3,2],[1, 4,1],[1,5,0],
...
[6,0,0]]

---

решено, и возможная комбинация рассчитывается как

factorial(given_num+n-1)/(factorial(given_num) * factorial(n-1))

Answer 1

from itertools import product
import numpy as np

tuples = []
given_num = 6
n = 3

for numbers in product(np.arange(given_num+1), repeat=n):

    if sum(numbers) == given_num:
        tuples.append(numbers)

print(tuples)

Answer 2

Если вы хотите повеселиться с рекурсивными функциями

def summands(num, N):
    if N==1:
        return [[num]]
    else:
        return [[i] + j for i in range(num+1) for j in summands(num - i, N-1)]

Answer 3

Здесь попробуйте использовать itertools product:

from itertools import product

given_num = 6
n = 3
all_comb = product(range(given_num+1), repeat=n)
final_lst = [i for i in all_comb if sum(i) == given_num]
print(final_lst)