從子表的總和小於給定數量

Question

列表生成子列表我有2個輸入爲：

p = 7 
s = [2 2 8 1 3] 

我知道如何itertool.combinations得到基本列表的子集，但要我想要的是連續的子列表，每個子元素的總和小於p。

所以結果應該是：

[2],[2,2],[2],[1],[1,3],[3] 

在這裏我得到了6個這樣的子列表，其中對於每個子列表，sum(sublist) < p。

Answer 1

試試這個：

p = 7 
s = [2,2,8,1,3] 
ans=[] 
for i in xrange(len(s)): 
    for j in xrange(i,len(s)): 
     if sum(s[i:j+1])<p: 
      ans.append(s[i:j+1]) 
print ans 

輸出：

[[2], [2, 2], [2], [1], [1, 3], [3]] 

Answer 2

如果你喜歡一個套，你可以試試這個：

>>> sum([[s[i:j+1] for j in range(i, len(s)) if sum(s[i:j+1])<p] for i in range(len(s))], []) 
[[2], [2, 2], [2], [1], [1, 3], [3]] 

的sum(list_of_lists, [])被壓扁列表創建的列表在列表理解中，如here所示。

或者你可以使用Julien Spronck在他的comment中提到的，這是遠遠優越的。