在pandas中，如何找到累積和大於閾值的行/索引？

Question

我想找到某一列中值的累積總和超過閾值的行（索引）。

我可以並且確實使用一個簡單的循環找到這個位置，如下所示：

def sum_to(df, col, threshold):
    s = 0
    for r in df.iterrows():
        if s + r[1][col] > threshold:
            return r[0]
        else:
            s += r[1][col]

    return len(df)

但是，我想知道在Pandas中是否有更好/更好的方法來實現這一目標。

Answer 1

最簡單的方法可能就是

df[col].cumsum().searchsorted(threshold)

但這假設您的列中沒有負數。

Answer 2

所以你想要這樣的東西：

df = pd.DataFrame({'A': [1, 2, 3, 4, 5]})
df[df['A'].cumsum() > 5]
#  A
#2 3
#3 4
#4 5