如何在Polars Rust中为现有列添加前后缀并生成新列？

我有一个包含A（i64类型）、B列的DataFrame，数据如下：

需要为A列的每个值添加前缀test_和后缀_new，生成名为TEST的新列，期望输出：

尝试使用apply函数处理但失败，代码如下：

let prefix = "prefix_";

    // Apply a closure to add the prefix to each value in the column
    let new_df = df.apply(|col: &Series| {
        // Ensure the column is of Int type
        if let Ok(int_col) = col.i32() {
            // Create a new Series with modified values
            let new_values = int_col
                .into_iter()
                .map(|value| format!("{}{}", prefix, value))
                .collect::<Vec<_>>();
            Ok(Series::new(new_values))
        } else {
            // Return the original column as-is for non-Int columns
            Ok(col.clone())
        }

报错信息：

the trait bound &Result<&mut polars::prelude::DataFrame, PolarsError>: IntoSeries is not satisfied

1. apply函数误用：df.apply用于批量处理DataFrame的所有列，但你仅需处理A列生成新列，完全不需要遍历所有列，反而引入不必要的复杂度。
2. 类型不匹配：你的A列是i64类型，但代码中用col.i32()做转换，类型不兼容会导致转换失败。
3. Result处理不完整：apply闭包的返回值处理逻辑有缺陷，且未正确处理整个调用的Result类型，最终导致 trait 绑定错误。

不需要使用apply，直接针对A列做字符串格式化即可，完整代码如下：

use polars::prelude::*;

fn main() -> Result<(), PolarsError> {
    // 构建原始DataFrame
    let df = df!(
        "A" => &[1i64, 3i64],
        "B" => &["value", "value"]
    )?;

    // 生成TEST列：拼接前缀和后缀
    let test_col = df
        .column("A")?
        .i64()?
        .into_iter()
        .map(|opt_val| opt_val.map(|val| format!("test_{}_new", val)))
        .collect::<Series>()
        .rename("TEST");

    // 将新列添加到原DataFrame
    let new_df = df.with_column(test_col)?;

    // 打印结果
    println!("{}", new_df);
    Ok(())
}

• df.column("A")?：安全获取A列，处理列不存在的错误；
• .i64()?：将列转换为i64类型的Series，与你的数据类型匹配；
• map(|opt_val| ...)：处理Polars中数值可能为Option的情况，避免空值引发panic；
• collect::<Series>()：将迭代器转换为Series并重命名为TEST；
• df.with_column(test_col)?：将新列添加到原DataFrame，返回新的DataFrame实例。

运行后输出的DataFrame与预期一致：

shape: (2, 3)
┌─────┬───────┬─────────────┐
│ A   ┆ B     ┆ TEST        │
│ --- ┆ ---   ┆ ---         │
│ i64 ┆ str   ┆ str         │
╞═════╪═══════╪═════════════╡
│ 1   ┆ value ┆ test_1_new  │
│ 3   ┆ value ┆ test_3_new  │
└─────┴───────┴─────────────┘

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