polars-rows-iter 0.13.0

Library for easy and convenient row iteration of polars dataframes
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45

#![allow(dead_code)]

use polars::prelude::*;
use polars_rows_iter::*;

mod column_names {
    pub const COLUMN_D: &str = "col_d";
}

#[derive(Debug, FromDataFrameRow)]
struct MyRow<A, B, C, D> {
    col_a: A,
    #[column("col_b")]
    b: B,
    col_c: C,
    #[column(column_names::COLUMN_D)]
    d: D,
}

fn create_dataframe() -> PolarsResult<DataFrame> {
    df!(
        "col_a" => [1i32, 2, 3, 4, 5],
        "col_b" => ["a", "b", "c", "d", "e"],
        "col_c" => ["A", "B", "C", "D", "E"],
        "col_d" => [Some(1.0f64), None, None, Some(2.0), Some(3.0)]
    )
}

fn run() -> PolarsResult<()> {
    let df = create_dataframe()?;

    let rows_iter = df.rows_iter::<MyRow<i32, &str, String, Option<f64>>>()?;

    for row in rows_iter {
        let row = row?;

        println!("{row:?}");
    }

    Ok(())
}

fn main() {
    run().unwrap()
}