RSI指标EMA三色信号线绘制逻辑故障求助

我在给RSI指标构建EMA线时，想用至少三种颜色过滤信号，但编写的if语句无法正常工作。参考其他脚本后猜测可能需要循环重绘解决，作为编程新手，求优化建议。

原代码：

switchcolor = color.yellow
//red
if (rsiema <= lenema[1]) and (mtmma <= mtmma[1])
    switchcolor := color.new(color.red, 100)
//green
if (rsiema >= lenema[1]) and (mtmma >= mtmma[1])
    switchcolor := color.new(color.green, 100)
//yellow
if  (rsiema <= lenema[1]) and (mtmma >= mtmma[1])
    switchcolor := color.new(color.yellow, 100)
if (rsiema > lenema[1]) and (mtmma < mtmma[1])
    switchcolor := color.new(color.yellow, 100)

plot(lenema, "RSI EMA", color = switchcolor, linewidth=2)

问题分析 #

1. 条件覆盖问题：多个if语句顺序执行，若某根K线同时满足多个条件，后执行的if会覆盖之前的颜色赋值，导致逻辑混乱。
2. 透明度设置错误：color.new(color.red, 100)中第二个参数是透明度（100为完全透明），这会导致线条直接看不见，也是你觉得逻辑不生效的核心原因之一。

解决方案 #

方案一：使用switch分支（Pine Script推荐写法）

switch会从上到下匹配第一个符合的条件，避免覆盖问题，逻辑更清晰：

//@version=5
indicator("RSI EMA Color Filter", overlay=false)

// 模拟指标变量（根据你的实际逻辑替换）
rsi = ta.rsi(close, 14)
rsiema = ta.ema(rsi, 9)
lenema = ta.ema(rsi, 18)
mtmma = ta.ema(rsi, 27)

switchcolor = color.yellow

switch true
    case (rsiema <= lenema[1]) and (mtmma <= mtmma[1])
        switchcolor := color.new(color.red, 0)  // 0为完全不透明
    case (rsiema >= lenema[1]) and (mtmma >= mtmma[1])
        switchcolor := color.new(color.green, 0)
    case (rsiema <= lenema[1]) and (mtmma >= mtmma[1]) or (rsiema > lenema[1]) and (mtmma < mtmma[1])
        switchcolor := color.new(color.yellow, 0)

plot(lenema, "RSI EMA", color=switchcolor, linewidth=2)

方案二：嵌套三元运算符（简洁写法）

适合分支逻辑简单的场景，代码更紧凑：

//@version=5
indicator("RSI EMA Color Filter", overlay=false)

// 模拟指标变量
rsi = ta.rsi(close, 14)
rsiema = ta.ema(rsi, 9)
lenema = ta.ema(rsi, 18)
mtmma = ta.ema(rsi, 27)

switchcolor = 
    (rsiema <= lenema[1]) and (mtmma <= mtmma[1]) ? color.new(color.red, 0) :
    (rsiema >= lenema[1]) and (mtmma >= mtmma[1]) ? color.new(color.green, 0) :
    color.new(color.yellow, 0)

plot(lenema, "RSI EMA", color=switchcolor, linewidth=2)

方案三：循环重绘（仅回溯历史信号时使用）

如果你的逻辑需要回溯所有K线重新计算颜色（普通实时信号过滤不需要），可以用循环+var数组实现：

//@version=5
indicator("RSI EMA Color Filter with Repaint", overlay=false)

// 模拟指标变量
rsi = ta.rsi(close, 14)
rsiema = ta.ema(rsi, 9)
lenema = ta.ema(rsi, 18)
mtmma = ta.ema(rsi, 27)

// 存储每根K线的颜色
var color[] colorHistory = array.new_color()

if barstate.islast
    array.clear(colorHistory)
    // 遍历所有历史K线计算颜色
    for i = 0 to bar_index
        rsiema_hist = ta.valuewhen(bar_index == i, rsiema, 0)
        lenema_prev_hist = ta.valuewhen(bar_index == i-1, lenema, 0)
        mtmma_hist = ta.valuewhen(bar_index == i, mtmma, 0)
        mtmma_prev_hist = ta.valuewhen(bar_index == i-1, mtmma, 0)
        
        currentColor = color.yellow
        if (rsiema_hist <= lenema_prev_hist) and (mtmma_hist <= mtmma_prev_hist)
            currentColor := color.new(color.red, 0)
        else if (rsiema_hist >= lenema_prev_hist) and (mtmma_hist >= mtmma_prev_hist)
            currentColor := color.new(color.green, 0)
        array.push(colorHistory, currentColor)

// 根据数组存储的颜色绘制线条
plot(lenema, "RSI EMA", color=array.size(colorHistory) > bar_index ? array.get(colorHistory, bar_index) : color.yellow, linewidth=2)

关键提示 #

• 调整透明度：color.new的第二个参数建议设为0-50之间，避免线条完全透明。
• 确保条件互斥：用switch或else if结构，避免同一K线匹配多个条件导致颜色被覆盖。
• 谨慎使用循环重绘：仅在需要回溯历史信号时使用，普通实时场景会增加脚本复杂度。

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