python定义长度为n的列表_在Python中生成长度N和为S的所有可能的列表

我正在尝试生成所有可能的长度N和S的列表。我已经编写了一些代码来这样做，但是对于任何大的(尤其是，我想要N=5，S=100)，我会遇到内存溢出错误。在

我在寻找一个更好的解决方案来解决这个问题，或者改进我的代码，这样我就可以在N=5，S=100上运行它。下面这两个程序协同工作，在嵌套列表中创建所有可能的数字组合，然后重新处理它们以使其成为正确的格式。下面复制了一些示例输出。在

我知道我的代码不是最好的。我是个工程师(我知道，我知道)，所以编码并不是我的专长。我很感激你能提供的任何帮助。在

编辑：我只是想澄清一些事情。首先，列表中的零是可以的，列表可以包含相同数字的倍数，并且列表中数字的顺序很重要。在def nToSum(N,S):

''' Creates a nested list of all possible lists of length N that sum to S'''

if N <= 1: #base case

return [S]

else:

L = []

for x in range(S+1): #create a sub-list for each possible entry of 0 to S

L += [[x,nToSum(N-1,S-x)]] #create a sub-list for this value recursively

return L

def compress(n=[],L): #designed to take in a list generated by nToSum

'&#