Polars批量替换ratio_前缀列inf为null的表达式异常问题

使用正则批量选择多列时，pl.when().then().otherwise() 表达式默认不会沿用原列名输出，因此会生成名为literal的新列，无法覆盖原有匹配列。

不需要循环遍历列，只需要在表达式末尾追加.name.keep()，指定输出保留匹配到的原列名，即可直接在原列上完成替换：

import polars as pl
import numpy as np

df = pl.DataFrame({"label":["a", "b", "c"], "ratio_a":[1.,2.,np.inf]})

df = df.with_columns(
    pl.when(pl.col("^ratio_\w+$").is_infinite())
    .then(None)
    .otherwise(pl.col("^ratio_\w+$"))
    .name.keep()
)

执行后不会生成多余列，所有以ratio_开头的列中的无穷值都会被替换为null。

更简洁的写法 #

针对单纯的值替换场景，可以直接用replace方法实现，代码更精简，默认保留原列名，还支持同时替换正、负无穷：

df = df.with_columns(
    pl.col("^ratio_\w+$").replace({
        np.inf: None,
        -np.inf: None
    })
)

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