Python3 - 总计为一个数字的 7 位数字的排列

Question

我需要在 Python3 中找到以下问题的解决方案。我试过 itertools.combinations 但不清楚如何做。

准备一个总和为 5 的 7 位数字。每个数字只能介于 0-4 之间。此外，还可以有重复。有效的示例编号是 -

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

如您所见，数字可能在此列表中出现多次。

如何创建满足上述条件的所有组合的列表？

Answer 1

你可以得到所有的总和为 5:

list(p for p in itertools.product(range(5),repeat = 7) if sum(p) == 5)

这会产生 455 个解决方案。