[已解决]求某数下方几个数字相加返回行索引怎么解决？

lovecpy

0      100    A
1      -20
2       30     B
3      -70
4      -50
5      500    A
6     -100
7     -420
8    -1000
9      600    A
10    -300
11     700   B
12    -300
13    -600
14    -200
15    -100

如上面得例子，索引0得值是100，如果 100+（-20）+30+（-70）+（-50）< 0 得时候 返回 -50得行索引4 如何得到4?
以此类推，500-100-420<0返回7  ，600-300+700-300-600-200< 0返回14  (仅当 第三列为字母A得时候才开始计算下方相加返回得索引)

最佳答案

月排行榜 / 总排行榜

逃兵

2021-1-27 09:23:28

import pandas as pd
a = {'num':[100,-20,30,-70,-50,500,-100,-420,-1000,600,-300,700,-300,-600,-200,-100],'level':['A','','B','','','A','','','','A','','B','','','','']}
df = pd.DataFrame(a)
for i in df.index:
        if df.level[i] == 'A':
                temp = df.num[i]
        elif df.level[i] != 'A':
                temp+=df.num[i]
        if temp<0:
                print(i)
                temp = 999999

跳转到最佳答案楼层

逃兵

import pandas as pd
a = {'num':[100,-20,30,-70,-50,500,-100,-420,-1000,600,-300,700,-300,-600,-200,-100],'level':['A','','B','','','A','','','','A','','B','','','','']}
df = pd.DataFrame(a)
for i in df.index:
        if df.level[i] == 'A':
                temp = df.num[i]
        elif df.level[i] != 'A':
                temp+=df.num[i]
        if temp<0:
                print(i)
                temp = 999999

lovecpy

谢谢，非常棒！