查看 rust-polars LazyFrame 列中的下一个值，同时仍在处理当前值

Answer 1

好的，我找不到用 LazyFrame 做所有事情的方法，因此我将 LazyFrame 转换为一个 Eager DataFrame，并且能够同时处理两列。

所以它现在工作。也许有人可以帮助我们实现仅使用 LazyFrame 的解决方案。

这是工作代码：

use polars::export::arrow::temporal_conversions::date32_to_date;

use polars::prelude::*;

fn main() -> Result<()> {
    let days = df!(
        "date_string" => &["1900-01-01", "1900-01-02", "1900-01-03", "1900-01-04", "1900-01-05",
        "1900-01-06", "1900-01-07", "1900-01-09", "1900-01-10"])?;

    let options = StrpTimeOptions {
        date_dtype: DataType::Date,   // the result column-datatype
        fmt: Some("%Y-%m-%d".into()), // the source format of the date-string
        strict: false,
        exact: true,
    };

    // convert date_string into dtype(date) and put into new column "date_type"
    // we convert the days DataFrame to a LazyFrame ...
    // because in my real-world example I am getting a LazyFrame
    let mut new_days_lf = days.lazy().with_column(
        col("date_string")
            .alias("date_type")
            .str()
            .strptime(options),
    );

    // Getting the weekday as a number:
    // This is what I wanted to do ... but I get a string result .. need u32
    // let o = GetOutput::from_type(DataType::Date);
    // new_days_lf = new_days_lf.with_column(
    //     col("date_type")
    //         .alias("weekday_number")
    //         .map(|x| Ok(x.strftime("%w").unwrap()), o.clone()),
    // );

    // This is the convoluted workaround for getting the weekday as a number
    let o = GetOutput::from_type(DataType::Date);
    new_days_lf = new_days_lf.with_column(col("date_type").alias("weekday_number").map(
        |x| {
            Ok(x.date()
                .unwrap()
                .clone()
                .into_iter()
                .map(|opt_name: Option<i32>| {
                    opt_name.map(|datum: i32| {
                        // println!("{:?}", datum);
                        date32_to_date(datum)
                            .format("%w")
                            .to_string()
                            .parse::<u32>()
                            .unwrap()
                    })
                })
                .collect::<UInt32Chunked>()
                .into_series())
        },
        o,
    ));

    new_days_lf = new_days_lf.with_column(
        col("weekday_number")
            .shift_and_fill(-1, 9999)
            .alias("next_weekday_number"),
    );

    // now we convert the LazyFrame into a normal DataFrame for further processing:
    let mut new_days_df = new_days_lf.collect()?;

    // convert the column to a series
    // to get a column by name we need to collect the LazyFrame into a normal DataFrame
    let col1 = new_days_df.column("weekday_number")?;

    // convert the column to a series
    let col2 = new_days_df.column("next_weekday_number")?;

    // now I can use series-arithmetics
    let diff = col2 - col1;

    // create a bool column based on "element == 2"
    // add bool column to DataFrame
    new_days_df.replace_or_add("weekday diff eq(2)", diff.equal(2)?.into_series())?;

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

    Ok(())
}