Rust Polars升级后DataFrame过滤报错:多谓词歧义问题求助
Polars Rust 多列筛选条件的修正方案
问题背景
升级至最新版Polars后,原用于筛选CK和CP列值均大于0的代码抛出歧义错误,原代码及报错信息如下:
原代码
let e11_filter = df .lazy() .filter(cols(["CK", "CP"]).gt(lit(0))) .collect()?;
报错信息
Error: ComputeError(ErrString("The predicate passed to 'LazyFrame.filter' expanded to multiple expressions: [(col(\"CK\")) > (0)], [(col(\"CP\")) > (0)], This is ambiguous. Try to combine the predicates with the 'all_exprs' or 'any_exprs' expression. Error originated just after this operation: DF [\"W\", \"Dataset\", \"MD\", \"CK\"]; PROJECT */8 COLUMNS; SELECTION: \"None\""))
报错原因
新版Polars对filter方法的参数做了严格限制:不再允许直接传入多列生成的多个独立表达式,必须明确指定条件的组合逻辑(所有条件满足/任一条件满足),避免语义歧义。
解决方法
由于需求是CK和CP列值均大于0(即所有条件同时成立),可通过以下两种方式修正:
方式1:使用all_exprs组合独立条件
use polars::prelude::*; let e11_filter = df .lazy() .filter(all_exprs([col("CK").gt(lit(0)), col("CP").gt(lit(0))])) .collect()?;
方式2:使用all_horizontal简化多列条件
利用cols批量选择列后,通过all_horizontal()对每行的多列结果做逻辑与(AND)运算:
use polars::prelude::*; let e11_filter = df .lazy() .filter(cols(["CK", "CP"]).gt(lit(0)).all_horizontal()) .collect()?;
若需求为任一列值大于0,只需将
all_horizontal()替换为any_horizontal()即可。
内容的提问来源于stack exchange,提问作者humphreylee
相关产品推荐
相关产品推荐

