在列表中添加项目，直到其总和超过阈值

Question

我要从值列表中创建一个新的值列表，直到它们加起来。

我是Python的新手，但我相信最好使用while循环。

L = [1,2,3,4,5,6,7,8,9]
i = 0
s = 0
while i < len(L) and s + L[i] < 20:
    s += L[i]
    i += 1

Answer 1

numpy 数组使这个变得简单

import numpy as np

arr = np.array(L)
arr[arr.cumsum() <= 20].tolist()
#[1, 2, 3, 4, 5]