Início / coding / Perguntas / 79544355
Accepted
Nyssance
Asked: 2025-03-30 16:47:30 +0800 CST

Como obter todos os grupos em polares por ferrugem?

  • 772

Em python, assim

df = pl.DataFrame({"foo": ["a", "a", "b"], "bar": [1, 2, 3]})
for name, data in df.group_by("foo"):  
    print(name)
    print(data)

saída:

(a,)
shape: (2, 2)
┌─────┬─────┐
│ foo ┆ bar │
│ --- ┆ --- │
│ str ┆ i64 │
╞═════╪═════╡
│ a   ┆ 1   │
│ a   ┆ 2   │
└─────┴─────┘
(b,)
shape: (1, 2)
┌─────┬─────┐
│ foo ┆ bar │
│ --- ┆ --- │
│ str ┆ i64 │
╞═════╪═════╡
│ b   ┆ 3   │
└─────┴─────┘

Como fazer o mesmo em Rust?

rust

1 respostas

  1. Best Answer

    Isso funciona um pouco diferente em Rust, porque um DataFrame / preguiçoso LazyGroupByé usado aqui.
    Você pode fazer algo assim usando a applyfunção:

    use polars::prelude::*;

fn main() -> PolarsResult<()> {
    let df = df!(
        "foo" => &["a", "a", "b"],
        "bar" => &[1, 2, 3]
    )?;

    let groups = df.group_by(["foo"])?;
    groups.apply(|group| {
        let group_name =
            group.column("foo")?.str()?.get(0).ok_or_else(|| {
                PolarsError::ComputeError("Failed to retrieve group name.".into())
            })?;
        println!("{}", group_name);
        println!("{:?}", group);
        Ok(group)
    })?;

    Ok(())
}
    • 4

relate perguntas