如果列表中的最后n个元素大于一个数字，则返回列表中的剩余项

Question

给定数字列表：

ar =[1,2,3,4,5,6,7,8,9,10]

如果最后的n个数字之和大于给定的数字，我需要找到列表的其余部分。 例如，如果给定数字为25，则返回的“剩余”列表将为[1,2,3,4,5,6,7] ，因为最后3个元素的总和为27。

我有一些代码，但是我很霸道地怀疑它效率很低，否则，鉴于任务的简单性质，它过于冗长。

def kp_find(ar, level):

    kp1=[]    
    for k in range(len(ar)): #this is required as in reality there is some
        kp1.append(k)        #work to be done within this loop 

    kp_fin = sorted(kp1,reverse = True)

    kp2=[]
    for k in kp_fin:
        kp2.append(k)
        sm_it = sum([i for i in kp2])
        if sm_it>level:
            place = [i for i, j in enumerate(kp_fin) if j == k]
            break
        else:
            place = []

    if place:
        return kp_fin[place[0]:len(kp1)]

print kp_find(ar,24)

如您所见，对于如此微不足道的东西，有很多代码。 任何帮助欢迎。

Answer 1

对于一个简单的任务，似乎确实有很多代码。 您可以从列表的右端弹出元素，直到满足条件为止。 像这样：

>>> def kp_find(ar, level):
...     result = ar[:]
...     sum_ = 0
...     while sum_ < level:
...         sum_ += result.pop()
...     return result
... 
>>> kp_find([1,2,3,4,5,6,7,8,9,10], 25)
[1, 2, 3, 4, 5, 6, 7]