修改Pine Script实现从指定日期起计算多标的最高价

核心思路 #

通过input.time()创建用户可自定义的起始日期输入控件，利用ta.barssince()定位到起始日期对应的K线位置，之后仅计算该位置之后的历史最高价（ATH），替代原全时段计算逻辑。

具体代码修改步骤 #

1. 添加起始日期输入控件
   在代码开头的indicator定义后，加入用户输入项：

//@version=5
indicator("ATH Distance Heatmap", overlay=false)

// 新增：用户自定义ATH计算起始日期
startDate = input.time(timestamp("2024-01-01 00:00"), title="ATH计算起始日期", tooltip="仅从此日期开始计算历史最高价")

2. 定位起始K线并计算限定时段ATH
   替换原全时段ATH计算代码，改为仅计算起始日期之后的最高价：

// 标记起始日期之后的K线
isAfterStart = time >= startDate

// 计算起始日期后的最高价：ta.barssince(isAfterStart)返回从起始K线到当前的周期数，+1确保包含起始K线
periodHigh = ta.highest(close, ta.barssince(isAfterStart) + 1)

// 处理起始日期之前的K线：若未到起始日期，将ATH设为当前收盘价（或根据需求设为na）
ath = na(periodHigh) ? close : periodHigh

3. 适配热力图计算逻辑
   确保后续的ATH距离（如当前价与ATH的百分比差值）仅基于限定时段的ATH计算，替换原代码中调用全时段ATH的部分即可。

完整修改示例片段 #

//@version=5
indicator("ATH Distance Heatmap", overlay=false, shorttitle="ATH Heatmap")

// 用户输入起始日期
startDate = input.time(timestamp("2024-01-01 00:00"), title="ATH计算起始日期")
isAfterStart = time >= startDate

// 计算限定时段的ATH
periodHigh = ta.highest(close, ta.barssince(isAfterStart) + 1)
ath = na(periodHigh) ? close : periodHigh

// 计算ATH距离百分比
athDistancePercent = ((close - ath) / ath) * 100

// 绘制热力图（示例）
heatmapColor = color.rgb(255, math.round(255 - (athDistancePercent * 2)), math.round(0 + (athDistancePercent * 2)))
bgcolor(heatmapColor, transp=80)

// 打印ATH值
plot(ath, color=color.new(color.red, 0), title="限定时段ATH")

关键注意事项 #

• input.time()的默认时间戳格式为timestamp("YYYY-MM-DD HH:MM")，可根据时区调整
• 若起始日期设置在未来，ta.barssince(isAfterStart)会返回na，此时通过na(periodHigh) ? close : periodHigh避免图表报错
• 多标的场景下，该逻辑会自动适配每个标的的起始日期后数据，无需额外修改多标的遍历代码

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