rust - 我如何在 Polars Rust 中的日期列的年份或工作日进行分组

Question

我正在尝试按日期列的年份对数据框进行分组。首先，让我们创建一个数据框:

let s0 = Series::new("date", &["2021-01-14","2022-04-09","2021-06-24","2022-12-04","2022-11-25"]);
let s1 = Series::new("values", &[1, 2, 3, 4, 5]);
let mut df = DataFrame::new(vec![s0, s1])?;
df.try_apply("date", |col_series| {Ok(col_series.utf8().unwrap().as_date(Some("%Y-%m-%d")).unwrap().into_series())});
let lf = df.lazy();

然后是我想要实现的(非工作)代码。

lf.groupby([col("date").year()]).agg([col("values").sum()]).collect()

Answer 1

我们可以通过调用 col("date").dt() 转到我们的 "date" 列的日期命名空间。

这使得 year() 函数可用。在您的情况下，正确的代码是

lf.groupby([col("date").dt().year()])
        .agg([col("values").sum()])
        .collect();

这取自 python 文档中的过滤部分

https://pola-rs.github.io/polars-book/user-guide/dsl/groupby.html#filtering