计算pandas datafram中满足一定求和条件的行数

Question

我试图获得pandas dataframe中某列的前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时，行数是3。在

谢谢

Answer 1

cumsum+idxmax应该可以工作：

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

^{pr2}$