计算满足熊猫数据框中某些求和条件的行数

Question

当pandas数据框中的某些列中的前x行的总值超过某些值时，我尝试获取行数。 我已经阅读了几种解决方案，但并非完全符合我的要求。 基本上，我可以使用以下代码所示的循环来执行此操作。 我只是想知道是否在python中有任何突击队做到这一点而没有循环？

import pandas as pd

df = pd.DataFrame({'A': pd.Series(range(1, 10), index = \
                  range(1, len(range(1,10))+1))})
Count = 0

for i in df.loc[:, 'A']:
    Count += i
    if (Count > 5):
        break

print('Row index:', i)

这段代码为我提供了我希望当A列中x第一行的总和超过5时的行数。

谢谢

Answer 1

cumsum + idxmax应该可以工作：

df.A.cumsum().gt(5).idxmax()

3