如何将条件计算列添加到rust中的polars数据帧？

[dependencies]
polars = "0.17"

use polars::prelude::*;

fn main() {
    // 创建数据帧
    let df = DataFrame::new(vec![
        ("A", &[1, 2, 3, 4, 5]),
        ("B", &["a", "b", "c", "d", "e"]),
    ])
    .unwrap();

    // 打印数据帧
    println!("{}", df);
}

use polars::prelude::*;

fn main() {
    // 创建数据帧
    let df = DataFrame::new(vec![
        ("A", &[1, 2, 3, 4, 5]),
        ("B", &["a", "b", "c", "d", "e"]),
    ])
    .unwrap();

    // 添加条件计算列
    let new_df = df.with_column(
        df.column("A")
            .unwrap()
            .eq(2) // 定义条件，例如判断"A"列的值是否等于2
            .alias("is_2"), // 为新列起一个别名
    );

    // 打印新的数据帧
    println!("{}", new_df);
}