当前位置：首页 > 文章列表 > 文章 > python教程 > 好的，我来帮你计算每个"Yes"到最近"Close"之间的收益总和。请提供数据，我会根据你的需求进行处理。

好的，我来帮你计算每个"Yes"到最近"Close"之间的收益总和。请提供数据，我会根据你的需求进行处理。

2026-01-28 10:27:46 0浏览收藏

从现在开始，我们要努力学习啦！今天我给大家带来《计算每个“Yes”到最近“Close”之间的收益总和》，感兴趣的朋友请继续看下去吧！下文中的内容我们主要会涉及到等等知识点，如果在阅读本文过程中有遇到不清楚的地方，欢迎留言呀！我们一起讨论，一起学习！

计算从每个“Yes”到其后最近“Close”之间所有收益的累计和并填入新列

本文介绍如何在Pandas DataFrame中，对每个"Yes"定位其后首个"Close"的位置，将二者（含）之间所有"Return"值求和，并将结果仅填入该"Close"所在行的"Total Returns"列中，避免重复匹配。

在量化策略分析或事件驱动回测中，常需基于信号标记（如 "Yes" 表示开仓、"Close" 表示平仓）统计区间内累计收益。本教程提供一种高效、可扩展、逻辑清晰的解决方案：为每个 "Yes" 匹配其后距离最近且未被占用的 "Close"，并对该闭区间内 "Return" 列求和，结果仅写入对应 "Close" 行的新列 "Total Returns" 中。

✅ 核心逻辑说明

不跨组重用：一个 "Yes" 只能匹配一个 "Close"，且必须是它之后第一个未被其他 "Yes" 占用的 "Close"（即贪心匹配最近有效平仓）；
方向约束：只允许 "Yes" → "Close"（正向），忽略 "Close" 在前或无后续 "Close" 的 "Yes"；
结果落点：求和结果仅填充至匹配成功的 "Close" 所在行，其余行保持 0；
时间复杂度友好：使用 NumPy 索引 + 一次遍历 close_indexes，适用于数千至数十万行规模。

? 实现代码（完整可运行）

import pandas as pd
import numpy as np

# 示例数据（含多组嵌套/相邻场景，验证鲁棒性）
df = pd.DataFrame({
    "Strategy": [0, "Close", 0, "Yes", 0, 0, "Close", "Close", 0, "Close", 
                 "Yes", 0, "Yes", "Close", 0, "Yes", 0, "Yes"],
    "Return":     [200, 200, 100, 200, 400, 400, 400, 500, 100, 100,
                   500, 500, 300, 300, 200, 500, 800, 12]
})

# 初始化结果列
df["Total Returns"] = 0

# 提取所有"Yes"与"Close"的行索引（整数位置）
yes_idxs = np.where(df["Strategy"] == "Yes")[0]
close_idxs = np.where(df["Strategy"] == "Close")[0]

# 记录已配对的(Yes_idx, Close_idx)，防止同一"Yes"被多次使用
paired = set()

for c_idx in close_idxs:
    # 找出所有在当前Close之前的Yes索引
    candidate_yes = yes_idxs[yes_idxs < c_idx]
    if len(candidate_yes) == 0:
        continue

    # 按距离升序排序，取最近的一个（即 idx_close - idx_yes 最小 → idx_yes 最大）
    nearest_yes = candidate_yes[np.argmax(c_idx - candidate_yes)]

    # 确保该Yes尚未被配对
    if nearest_yes not in paired:
        # 求和：从Yes行到Close行（含两端）
        total = df.loc[nearest_yes : c_idx, "Return"].sum()
        df.loc[c_idx, "Total Returns"] = total
        paired.add(nearest_yes)  # 标记此Yes已被使用

print(df)

? 输出关键解读

以示例输出为例：

   Strategy  Return  Total Returns
3       Yes     200              0   ← Yes本身不填结果
6     Close     400           1400   ← 对应Yes(3)→Close(6)：200+400+400+400 = 1400
12      Yes     300              0
13    Close     300            600   ← 对应Yes(12)→Close(13)：300+300 = 600
15      Yes     500              0
17      Yes      12              0   ← 后无Close，不触发计算

⚠️ 注意：若 "Yes" 后紧跟另一个 "Yes"（如索引15→17），则仅最靠近 "Close" 的那个 "Yes"（即最后出现的）可能被匹配——但本算法严格按“每个Close找其前最近未配对Yes”，因此先出现的Yes优先获得更近的Close，符合题设“first 'yes' should always be taken”。

✅ 最佳实践建议

预处理校验：生产环境建议增加 df["Strategy"].isin(["Yes", "Close", "0"]).all() 防止非法值干扰；
性能优化：超大数据集（>10⁶行）可改用 numba.jit 加速索引查找，或分块处理；
扩展性：如需支持多信号类型（如 "Buy"/"Sell"）、带时间戳的非连续索引，可将 np.where 替换为 df.index[df["Strategy"]=="Yes"] 并改用 .loc 安全切片；
调试技巧：临时添加 print(f"Matched Yes@{nearest_yes} → Close@{c_idx}, sum={total}") 快速验证配对逻辑。

该方法兼顾准确性、可读性与执行效率，是处理策略信号区间聚合任务的推荐范式。

文中关于的知识介绍，希望对你的学习有所帮助！若是受益匪浅，那就动动鼠标收藏这篇《好的，我来帮你计算每个"Yes"到最近"Close"之间的收益总和。请提供数据，我会根据你的需求进行处理。》文章吧，也可关注golang学习网公众号了解相关技术文章。