Início / coding / Perguntas / 79463690
Accepted
Roger V.
Asked: 2025-02-24 21:28:16 +0800 CST

Renomeando uma única coluna com nome desconhecido em Polars-Rust

  • 772

O seguinte renomeia uma coluna em um dataframe Polars-Rust:

#![allow(unused_variables)]
use polars::prelude::*;

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

    println!("{:?}", df);

    let new_name = <PlSmallStr>::from_str("letters");
    let _ = df.rename("names", new_name);
    println!("{:?}", df);
}

Suponha agora que o nome da primeira coluna seja desconhecido (por exemplo, o dataframe é lido de um arquivo csv/excel, onde não tem nome) e eu gostaria de renomeá-lo para uso futuro:

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

    println!("{:?}", df);

    let old_name = df.get_column_names_str()[0];
    // let old_name = df.get_column_names_str()[0].clone();
    // let old_name = &mut df.get_column_names_str()[0].clone();
    // let mut old_name = &mut df.get_column_names_str()[0].clone();
    let new_name = <PlSmallStr>::from_str("letters");
    let _ = df.rename(old_name, new_name);
    println!("{:?}", df);
}

Isso resulta em um erro

error[E0502]: cannot borrow `df` as mutable because it is also borrowed as immutable
  --> src/main.rs:32:13
   |
27 |     let old_name = df.get_column_names_str()[0];
   |                    -- immutable borrow occurs here
...
32 |     let _ = df.rename(old_name, new_name);
   |             ^^^------^^^^^^^^^^^^^^^^^^^^
   |             |  |
   |             |  immutable borrow later used by call
   |             mutable borrow occurs here

For more information about this error, try `rustc --explain E0502`.
error: could not compile `rename_col` (bin "rename_col") due to 1 previous error

Está mais ou menos claro por que o erro ocorre, mas não está claro como corrigi-lo...
As linhas comentadas mostram minhas várias tentativas malsucedidas de resolver esse problema.

Relacionado:
Em Rust, como renomear todas as colunas de um Polars Dataframe?
Como renomear nomes de colunas com a primeira linha em polars?

dataframe

2 respostas

  1. Best Answer

    Alternativamente, você pode usar df.get_column_names_owned()a função que retornará Vec<PlSmallStr>.

    let column_names = df.get_column_names_owned();
let old_name = column_names.first().unwrap();
let new_name = <PlSmallStr>::from_str("letters");
let _ = df.rename(old_name, new_name);

    OU

    let old_name = &df.get_column_names_owned()[0];
let new_name = <PlSmallStr>::from_str("letters");
let _ = df.rename(old_name, new_name);
println!("{:?}", df);
    • 2

  2. Os seguintes trabalhos (graças a uma dica de @cafce25):

    #![allow(unused_variables)]
use polars::prelude::*;

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

    println!("{:?}", df);

    let old_name = df.get_column_names_str()[0].to_owned();
    let new_name = <PlSmallStr>::from_str("letters");
    let _ = df.rename(&old_name, new_name);
    println!("{:?}", df);
}
    • 1

relate perguntas