如何对Polars Series集合执行逻辑运算并生成结果序列？

针对你遇到的Polars无直接Series级别逻辑运算的问题，以下是几种可行的实现方案：

方法一：使用表达式API（推荐） #

Polars的表达式系统原生支持多列（Series）的元素级逻辑运算，适合批量处理场景，代码简洁且性能最优：

use polars::prelude::*;

#[test]
fn test_series_logical_expr() {
    let s1 = Series::new("bool_series1", &[true, false, true, false]);
    let s2 = Series::new("bool_series2", &[true, true, false, false]);
    let s3 = Series::new("bool_series3", &[true, false, false, false]);

    // 将Series装入DataFrame统一处理
    let df = DataFrame::new(vec![s1, s2, s3]).unwrap();

    // 示例：所有列元素同时为true时结果为true（逻辑与）
    let result = df
        .select([
            col("bool_series1")
                .and(col("bool_series2"))
                .and(col("bool_series3"))
                .alias("all_true_result")
        ])
        .unwrap();

    println!("{}", result);
}

方法二：转换为BooleanChunked手动运算 #

如果需要直接操作独立Series，可以把Series转换为BooleanChunked类型，利用其提供的逻辑运算方法：

use polars::prelude::*;

#[test]
fn test_series_logical_manual() {
    let s1 = Series::new("bool_series1", &[true, false, true, false]);
    let s2 = Series::new("bool_series2", &[true, true, false, false]);
    let s3 = Series::new("bool_series3", &[true, false, false, false]);

    // 将Series转为BooleanChunked类型
    let bc1 = s1.bool().unwrap();
    let bc2 = s2.bool().unwrap();
    let bc3 = s3.bool().unwrap();

    // 逐步执行逻辑运算（这里以逻辑或为例）
    let temp = bc1.or(bc2).unwrap();
    let result_bc = temp.or(bc3).unwrap();

    // 转回Series类型
    let result_series = result_bc.into_series();
    println!("Result Series: {:?}", result_series);
}

方法三：批量处理通用函数 #

如果需要处理数量不固定的布尔Series，可以封装一个通用函数批量运算：

use polars::prelude::*;

fn logical_or_all(series_list: &[Series]) -> Result<Series, PolarsError> {
    if series_list.is_empty() {
        return Err(PolarsError::NoData("未传入任何Series".into()));
    }

    // 初始化结果为第一个Series的BooleanChunked
    let mut result_bc = series_list[0].bool()?.clone();

    // 遍历剩余Series执行逻辑或
    for s in series_list.iter().skip(1) {
        let bc = s.bool()?;
        result_bc = result_bc.or(bc)?;
    }

    Ok(result_bc.into_series())
}

#[test]
fn test_batch_logical_ops() {
    let s1 = Series::new("bool_series1", &[true, false, true, false]);
    let s2 = Series::new("bool_series2", &[true, true, false, false]);
    let s3 = Series::new("bool_series3", &[true, false, false, false]);

    let series_vector = vec![s1, s2, s3];
    let result_series = logical_or_all(&series_vector).unwrap();

    println!("Result Series: {:?}", result_series);
}

注意事项 #

• 所有参与运算的Series必须长度一致，否则Polars会抛出维度不匹配错误
• 除了and/or，还可以使用not方法实现取反逻辑，表达式API还支持更复杂的条件组合
• 表达式API支持链式调用，可结合过滤、分支判断等操作实现复杂逻辑

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