python - 在 Python 中列出具有任意数量变量的理解？

Question

我要解决的问题是获取集合 S，它包含 X 个整数的所有可能排列，使得所有整数都大于 0，并且在添加时等于某个 Y。

所以，如果 X 等于 2 且 Y 等于 5，则可以使用以下列表推导式解决问题:

[(a,b) for a in range(1,5) for b in range(1,5) if a+b==5]

但是，我正在尝试求解任意 X 和任意 Y。我可以将其编写为一系列巨大的 if 语句，但我觉得我已经非常接近一个漂亮、干净的列表理解来处理它。有什么方法可以用列表理解来做到这一点吗？

Answer 1

类似的东西:

[x for x in itertools.product(*variables) if sum(x)==5]

其中变量是您的范围的列表/元组