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

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

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

复制代码

跳转到最佳答案楼层

逃兵

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

复制代码

lovecpy

谢谢，非常棒！