Pandas提取符合特定条件的负值序列问题排查

核心问题分析 #

你的代码只提取了单个元素，是因为错误地孤立判断每个元素是否满足「前后值小于当前负值」——这个逻辑实际是在找局部最大值（数值上更大的负值），但你的真实需求是提取连续单调递减的负值序列（即序列内元素依次变小，越来越负），这和预期输出匹配。

修正后的代码 #

假设你的原始DataFrame构造如下（基于预期输出示例）：

import pandas as pd

# 示例输入数据
data = {'Value': [0.5, -0.1, -0.2, -0.3, -0.4, 0.3, -0.15, -0.25, 0.1]}
df = pd.DataFrame(data)

用向量化操作替代遍历，精准提取目标序列：

# 1. 标记关键特征
df['is_negative'] = df['Value'] < 0
# 生成连续负值块的ID（非负值作为块分隔符）
df['block_id'] = (~df['is_negative']).cumsum()
# 标记当前元素是否比前一个元素小（递减）
df['is_decreasing'] = df['Value'].diff() < 0

# 2. 筛选符合条件的负值块：全为负值且块内单调递减
valid_blocks = df.groupby('block_id').filter(
    lambda x: x['is_negative'].all() and x['is_decreasing'].iloc[1:].all()
)

# 3. 提取完整序列
result = [group['Value'].tolist() for _, group in valid_blocks.groupby('block_id')]

print(result)
# 输出：[[-0.1, -0.2, -0.3, -0.4], [-0.15, -0.25]]

修正逻辑说明 #

1. 划分连续负值块：用非负值作为分隔符，把连续的负值区域分成独立块，避免跨非负值区域提取。
2. 验证递减趋势：对每个负值块，检查除第一个元素外，所有后续元素是否都比前一个小（保证单调递减）。
3. 提取完整序列：把符合条件的整块负值序列收集起来，而不是单个元素。

内容的提问来源于stack exchange，提问作者Divyank