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

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]

Answer 2

由于您标记了熊猫：

pd.Series(L, index=np.cumsum(L)).loc[:20].values

输出：

array([1, 2, 3, 4, 5], dtype=int64)