Polars重构RSI计算表达式 拆分中间变量不生效问题咨询

你拆分公共逻辑的思路完全可行，写法无法运行基本是链式调用优先级错误、变量作用域问题导致的，以下是完全符合「返回纯Polars表达式、无需上下文求值」要求的实现，同时做了逻辑精简：

核心实现 #

首先抽离重复计算的公共部分，再拆分上涨、下跌收益的滚动均值，最终组合成RSI表达式即可，所有中间变量都是pl.Expr类型，组合后依然是合法的Polars表达式：

import polars as pl

def rsi_expr(window: int) -> pl.Expr:
    # 抽离公共涨跌幅计算，避免重复运算
    delta = pl.col("close").pct_change()
    # U：周期内平均上涨收益
    U = (
        pl.when(delta >= 0)
        .then(delta)
        .otherwise(0.0)
        .rolling_mean(window_size=window)
    )
    # V：周期内平均下跌损失（转成正数值，用-delta代替abs效率更高）
    V = (
        pl.when(delta < 0)
        .then(-delta)
        .otherwise(0.0)
        .rolling_mean(window_size=window)
    )
    return (100 * U / (U + V)).alias(f"rsi_{window}")

常见踩坑说明 #

之前拆分后无法运行，基本是以下几个原因：

• 链式调用优先级错误：换行时没有用括号包裹整个链式表达式、或者漏了行续符，导致.rolling_mean()、.abs()等方法没有作用在整个when-then-otherwise的输出上，而是错误绑定到了otherwise(0.0)的单步结果上，引发计算逻辑异常。
• 循环变量绑定问题：如果在循环里批量生成不同周期的RSI表达式，直接引用循环变量window会触发Python晚绑定问题，所有表达式最终会用最后一个循环的window值，封装成带参数的函数即可彻底规避这个问题。
• 下跌收益逻辑错误：部分写法把abs()放在otherwise外层，虽然逻辑上等价，但如果括号位置写错，会出现把0值做无意义转换、漏算下跌幅度的问题，直接在then分支用-delta把负收益转成正数，逻辑更清晰也更高效。

使用方式 #

这个函数返回的是纯Polars表达式，可以直接传入任何支持表达式的Polars接口，不需要提前绑定DataFrame：

# 示例：计算14周期RSI
df = df.with_columns(rsi_expr(window=14))

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