df-derive 0.3.0

Derive fast conversions from Rust structs into 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
46
47
48
49
50
51

use criterion::{Criterion, criterion_group, criterion_main};
use df_derive::ToDataFrame;

#[path = "support/mod.rs"]
mod bench_support;
#[path = "../tests/common.rs"]
mod core;
use crate::bench_support::configure_criterion;
use crate::core::dataframe::ToDataFrameVec;

const N_HOSTS: usize = 10_000;

#[derive(ToDataFrame, Clone)]
struct Metric {
    name: String,
    value: f64,
}

#[derive(ToDataFrame, Clone)]
struct Host {
    id: u64,
    metrics: Vec<Metric>,
}

fn generate_hosts() -> Vec<Host> {
    // All hosts have empty metrics vectors; ensures List columns get empty inner series per row
    (0..N_HOSTS)
        .map(|i| Host {
            id: i as u64,
            metrics: Vec::new(),
        })
        .collect()
}

fn benchmark_empty_nested_vec(c: &mut Criterion) {
    let hosts = generate_hosts();

    c.bench_function("empty_nested_vec_conversion", |b| {
        b.iter(|| {
            let df = std::hint::black_box(&hosts).to_dataframe().unwrap();
            std::hint::black_box(df)
        });
    });
}

criterion_group! {
    name = benches;
    config = configure_criterion();
    targets = benchmark_empty_nested_vec
}
criterion_main!(benches);