使用when->then执行Polars聚合触发警告，是否存在风险？

这个警告必须重视，存在两个核心风险：

• 后续版本兼容性问题：Polars明确标注该行为属于「实验性特性」，意味着官方随时可能修改逻辑——比如当前返回包含匹配值的列表，后续版本可能改成只返回第一个匹配值、自动过滤null，甚至直接禁止这种写法，届时你的代码会出现报错或输出不符合预期的结果。
• 结果逻辑的不确定性：警告提到「生成的行数可能与分组操作预期不符」，举个实际场景：如果某个a分组里有3行数据，其中2行b=1、1行b=2，当前写法会生成[1,1,null]的列表；但常规聚合逻辑（如sum）只会返回单个值，这种结构差异很容易让后续依赖列表长度、内容的代码出现逻辑bug。

既然你需要「即使结果为null也要保留对应值」的逻辑，推荐换成更明确的、非实验性的写法，彻底消除警告和风险：

# 写法1：用map_elements明确处理每个元素
df.group_by("a").agg(
    pl.col("b").map_elements(lambda val: val if val == 1 else None).alias("b")
)

# 写法2：用where保留原位置的null
df.group_by("a").agg(
    pl.col("b").where(pl.col("b") == 1).alias("b")
)

这两种写法的行为和你的需求完全匹配，且是Polars稳定支持的特性，不会出现版本变更导致的问题。

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