本文介绍如何在 polars 中动态为表达式（expr）中的原始列名（root_names）和别名添加后缀，重点解析基于 ast 序列化/反序列化的临时方案及其局限性，并强调官方不推荐该方式的原因。
本文介绍如何在 polars 中动态为表达式（expr）中的原始列名（root_names）和别名添加后缀，重点解析基于 ast 序列化/反序列化的临时方案及其局限性，并强调官方不推荐该方式的原因。
在 Polars 的表达式系统中，expr.meta.root_names() 可以便捷获取表达式所依赖的原始列名（如 "A" 和 "B"），但 Polars 当前（v1.0+）并未提供公开、稳定、面向用户的表达式结构重写（expression rewriting）API。这意味着你无法直接调用类似 expr.rename_columns(...) 或 expr.transform_root_names(...) 的方法来批量修改底层列引用或别名。
不过，一个技术上可行（但需谨慎使用）的绕过方案是利用 Polars 内部的表达式序列化机制：通过 .meta.serialize(format="json") 将表达式转为 JSON 格式的抽象语法树（AST），手动修改其中的 "Column" 和 "Alias" 字段，再通过 pl.Expr.deserialize() 重建表达式。例如：
import json
import polars as pl
def suffix_all(expr, suffix: str) -> pl.Expr:
def _add_suffix(obj):
if isinstance(obj, dict):
if "Column" in obj:
obj["Column"] = obj["Column"] + suffix
if "Alias" in obj and len(obj["Alias"]) >= 2:
# Alias: [inner_expr, alias_name]
obj["Alias"][-1] = obj["Alias"][-1] + suffix
return obj
ast_json = expr.meta.serialize(format="json")
modified_ast = json.loads(ast_json, object_hook=_add_suffix)
return pl.Expr.deserialize(json.dumps(modified_ast).encode(), format="json")
# 使用示例
df = pl.DataFrame({"A_suffix": [2, 7, 3], "B_suffix": [10, 7, 1]})
expr = pl.col("A").dot(pl.col("B")).alias("AdotB")
result = df.with_columns(suffix_all(expr, "_suffix"))
print(result)
该方案在简单场景下可正常工作，输出结果包含正确重命名的列（A_suffix, B_suffix, AdotB_suffix）及计算值。但必须注意以下关键限制：
- ⚠️ 版本不稳定：文档明确警告
serialize的输出格式“not stable across Polars versions”。任意小版本升级都可能导致 JSON 结构变更，使反序列化失败或产生不可预期行为； - ⚠️ 非公开契约：序列化 AST 属于内部实现细节，不在公共 API 保证范围内，未来可能被移除或重构；
- ⚠️ 语义风险：手动修改 AST 不校验列是否存在、类型是否兼容，也跳过 Polars 的表达式验证逻辑，易引入静默错误；
- ✅ 适用场景仅限调试与实验：适合临时脚本、测试探索或内部工具开发，绝不建议用于生产环境或长期维护代码。
更稳健的替代思路包括：
- 在构建表达式前，显式使用带后缀的列名（如
pl.col("A_suffix").dot(pl.col("B_suffix")).alias("AdotB_suffix")）； - 封装逻辑为函数，接受列名前缀/后缀参数，动态生成表达式；
- 对 DataFrame 预处理（如
df.rename(lambda c: c + "_suffix")），再基于新列名构造表达式。
总之，虽然序列化—修改—反序列化路径展示了 Polars 表达式结构的可操作性，但它是一把“双刃剑”。理解其原理有助于深入掌握 Polars 内部机制，但在工程实践中，请始终优先选择清晰、稳定、符合 API 设计意图的方式。
