Pandas Series去除连续重复值 保留买卖信号首次实例不删整行

核心需求是不删除整行、仅清除连续重复的buy/sell信号，只保留每段连续信号的第一个取值，直接通过行值对比即可实现，不需要做行删除操作。

实现代码 #

1. 核心逻辑 #

用shift()方法把signal列向下偏移一位，逐行对比当前行信号和上一行信号是否一致：

• 不一致：说明是新一段连续信号的起点，保留原信号值
• 一致：说明是连续重复的后续信号，将值置空即可

2. 可直接运行的示例 #

import pandas as pd
import numpy as np

# 测试用样例数据，可替换为你的实际DataFrame
df = pd.DataFrame({
    "trade_date": ["2024-01-01", "2024-01-02", "2024-01-03", "2024-01-04", "2024-01-05", "2024-01-06"],
    "signal": ["buy", "buy", "buy", "sell", "sell", "buy"]
})

# 处理逻辑：如果需要保留原始signal列就新建列存储结果，不需要的话可以直接赋值给df["signal"]覆盖原列
df["signal_processed"] = np.where(
    df["signal"] != df["signal"].shift(1),
    df["signal"],
    None  # 连续重复项置空，需要替换为空字符串/其他标记直接改这个参数即可
)

处理效果参考 #

处理前后的信号列对比如下：

处理前原始signal：
["buy", "buy", "buy", "sell", "sell", "buy"]

处理后signal_processed：
["buy", None, None, "sell", None, "buy"]

注意事项 #

• 整个处理过程不会修改DataFrame的行数量、其他列的取值、原有行索引，完全满足不删除整行的要求
• 如果你的signal列本身存在空值，可以先通过fillna()给空值设置临时占位符再做对比，避免空值干扰连续段判断
• 如果需要标记信号切换的位置，也可以把对比结果生成单独的标记列，方便后续回测逻辑调用

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