使用For循环提取相关性匹配Bar的Index及数组异常排查

核心问题拆解 #

• for循环逻辑失效：原代码中length = for i = offset to lookback + offset仅将length赋值为循环的最后一个i值，并未遍历范围内所有历史K线做相关性检查，导致只判断了单根目标K线，而非全部。
• bar index提取错误：变量赋值未与循环绑定，无法捕获符合条件的历史K线索引；且触发条件依赖错误的length值，记录的索引完全不准确。
• 数组异常原因：test_sim的触发逻辑混乱，导致数组push时机错误，可能在脚本初始化阶段（数组为空）调用array.get()报错，但后续数组有值后array.size()返回大于0，造成“矛盾”假象。

修正后的代码实现 #

//@version=5
indicator("相关性匹配追踪", overlay=true)

offset = 10 
lookback = 100
var bool corr_match = false
var int matched_bar_idx = na
var int[] matched_bar_array = array.new<int>(0)

// 示例：假设get_corr是已实现的相关性计算函数，这里用收盘价与开盘价的相关性模拟
get_corr = ta.correlation(close, open, 20)

if barstate.isnew
    // 重置本轮K线的匹配状态
    corr_match := false
    matched_bar_idx := na
    
    // 遍历指定范围的所有历史K线
    for i = offset to lookback + offset
        // 计算相关性差异值（同号取差值绝对值，异号取和值绝对值）
        corr_diff = (get_corr[i] > 0 and get_corr > 0) or (get_corr[i] < 0 and get_corr < 0) ? math.abs(get_corr[i] - get_corr) : math.abs(get_corr[i] + get_corr)
        
        // 匹配相似相关性范围
        if corr_diff <= 0.005
            matched_bar_idx := bar_index[i]
            corr_match := true
            // 若只需第一个匹配项，可添加break终止循环
            // break
        
        // 相关性差异过大时重置标记
        if corr_diff >= 0.05
            corr_match := false
    
    // 确认匹配有效时，将历史K线索引存入数组
    if corr_match and not na(matched_bar_idx)
        array.push(matched_bar_array, matched_bar_idx)

// 调试用：打印数组长度与最后一个匹配索引
if barstate.isnew
    label.new(bar_index, high, 
              str.tostring(array.size(matched_bar_array)) + " | " + 
              str.tostring(array.get(matched_bar_array, array.size(matched_bar_array)-1) if array.size(matched_bar_array) > 0 else na), 
              color=color.blue)

关键修改说明 #

• 遍历逻辑修正：将相关性检查放入for循环内部，逐一对比当前K线与每根历史K线的相关性，确保覆盖所有目标范围。
• 索引提取优化：在循环内满足匹配条件时，直接赋值matched_bar_idx为对应历史K线的bar_index[i]；若需收集所有匹配项，可直接在循环的if条件内执行array.push。
• 数组异常修复：
  1. 每次barstate.isnew时重置标记变量，避免历史状态干扰。
  2. 调用array.get()前先判断数组长度是否大于0，防止空数组访问报错。
  3. 仅在确认匹配有效时才执行数组push，避免无效元素存入。

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