Rust中如何重命名Polars DataFrame所有列？解决借用冲突

你遇到的错误是Rust借用检查器的典型限制：同一时间不能对同一个值同时持有不可变引用和可变引用。

错误原因 #

df.get_column_names()会返回指向原DataFrame列名的不可变引用，当你在迭代这些引用的过程中调用df.rename()时，需要对DataFrame进行可变借用——这直接违反了Rust的借用规则，因此报错。

解决方案1：先收集列名到独立集合 #

先把列名复制到一个独立的Vec中，脱离对原DataFrame的引用，之后再迭代这个独立集合进行重命名：

use polars::prelude::*;

fn main() {
    let mut df = df! [
        "names" => ["a", "b", "c"],
        "values" => [1, 2, 3],
    ].unwrap();

    let columns = vec![
        "name_chg",
        "value_chg",
    ];

    // 将列名复制到独立Vec，解除对df的引用依赖
    let old_columns: Vec<_> = df.get_column_names().to_vec();
    
    old_columns.into_iter().zip(columns).for_each(|(old, new)| {
        df.rename(old, new).unwrap();
    });

    println!("{df}");
}

解决方案2：使用批量重命名（更推荐） #

Polars提供了批量重命名的API，直接传入新旧列名的映射表，只需要一次可变借用即可完成操作，更高效也更符合Polars的使用习惯：

use polars::prelude::*;
use std::collections::HashMap;

fn main() {
    let mut df = df! [
        "names" => ["a", "b", "c"],
        "values" => [1, 2, 3],
    ].unwrap();

    // 构建新旧列名的映射
    let rename_map = [("names", "name_chg"), ("values", "value_chg")]
        .into_iter()
        .collect::<HashMap<_, _>>();
    
    // 批量重命名
    df.rename(&rename_map).unwrap();

    println!("{df}");
}

补充说明 #

对于无列名的数据集，你可以先通过df.set_column_names()给列设置临时名称，之后再用上述方法重命名；或者直接在读取数据时指定列名（比如用CsvReader的with_column_names方法），避免后续的重命名操作。

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