Polars自定义sigmoid函数如何保留原列名？

问题背景 #

实现的Sigmoid函数运行正常，但生成的列名被Polars自动命名为literal；改写表达式的方式虽能隐式保留列名，但写法不直观且依赖Polars内部的列名追踪逻辑；手动添加.alias()又无法适配批量处理场景。

解决方案 #

方法1：使用keep_name()强制保留原列名

Polars的Expr对象提供了keep_name()方法，可直接让计算后的表达式继承输入列的名称，无需手动指定：

import polars as pl

def sigmoid(c: pl.Expr) -> pl.Expr:
    return (1 / ((-c).exp() + 1)).keep_name()

使用时不管是单列还是批量处理，都能自动保留原列名：

# 单列处理
df.select(pl.col('a').pipe(sigmoid))
# 批量处理所有列
df.select(pl.all().pipe(sigmoid))

方法2：通过meta.output_name()获取列名后手动别名

如果需要更精细化的控制，可通过表达式的meta属性获取原列名，再用.alias()指定：

def sigmoid(c: pl.Expr) -> pl.Expr:
    return (1 / ((-c).exp() + 1)).alias(c.meta.output_name())

这种方式和keep_name()效果一致，但keep_name()更简洁直观。

说明 #

• 原写法生成literal的原因：当表达式从字面量（如1 / ...）开始计算时，Polars无法关联到原始列的名称，因此默认命名为literal。
• 避免使用改写表达式的隐式追踪方式：这种写法依赖Polars内部对列名的追踪逻辑，可读性差且在复杂表达式中可能失效。

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