如何提取DataFrame中首次出现的最后正行及后续行
解决方法
你的核心问题是没记录目标行的索引,只存了值,导致无法精准定位到目标行,也没法取到下一行。下面给两种实用的解决思路:
方法一:直接定位第一个负值的索引
我们要找的是首次出现的"最后一个正值+第一个负值",本质就是找第一个负值的位置,它的前一行就是目标正值行:
import pandas as pd # 构造示例数据 data = {'col1': [1,2,3,4,5,6,7,8,9], 'col2': [0.6,0.42,0.78,0.5,-0.33,-0.55,-4,0.6,0.5]} df = pd.DataFrame(data) # 获取第一个负值所在行的索引 first_neg_idx = df[df['col2'] <= 0].index[0] # 提取目标两行 df_new = df.loc[[first_neg_idx - 1, first_neg_idx]] print(df_new)
运行后就能得到你要的结果:
col1 col2 3 4 0.50 4 5 -0.33
方法二:改进你的循环逻辑
保留你原来的循环思路,但改成记录索引而非值,这样就能精准定位:
last_pos_idx = None # 遍历索引和对应的值 for idx, num in df['col2'].items(): if num <= 0: break last_pos_idx = idx # 提取目标行(最后一个正值行 + 下一行的负值行) df_new = df.loc[[last_pos_idx, last_pos_idx + 1]]
为什么你的原代码有问题?
你只保存了last_posnum这个值,但如果col2里有多个相同的正值(比如示例末尾的0.5),df.loc[df['col2'] == last_posnum]会把所有等于0.5的行都取出来,没法精准定位到首次出现的那段的最后一个正值行。用索引定位就不会有这个问题。
内容的提问来源于stack exchange,提问作者travgeol
相关产品推荐
相关产品推荐

