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

Question

I tried to get the number of rows when the total values of the first x rows in certain column in pandas dataframe exceed certain values. I've read several solutions, but not exactly what I am looking for. Basically I can do this with a loop as shown in the following codes. I just wondering whether there is any commando in python to do this without loop?

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)

this codes provides me what I want that the number of row when the sum of x first row in column A exceed 5 is three.

thanks

Answer 1

cumsum + idxmax应该可以工作：

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

3