plotters-statistical 0.1.0

Statistical chart primitives (box, violin, ROC, PR, regularization-path, residual) as native plotters series
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
52
53
54
55

//! Multi-group violin plot rendered to `violin_plot.svg`, shown both without and
//! with the embedded box-plot overlay (left / right panels).

use plotters::prelude::*;
use plotters_statistical::ViolinPlotSeries;

fn sample(center: f64, spread: f64, n: usize) -> Vec<f64> {
    // Deterministic pseudo-sample: a couple of sinusoidal "modes" around center.
    (0..n)
        .map(|i| {
            let t = i as f64 / n as f64;
            center + spread * ((t * std::f64::consts::TAU * 2.0).sin() + (t - 0.5) * 2.0)
        })
        .collect()
}

fn main() -> Result<(), Box<dyn std::error::Error>> {
    let root = SVGBackend::new("violin_plot.svg", (960, 500)).into_drawing_area();
    root.fill(&WHITE)?;
    let (left, right) = root.split_horizontally(480);

    let groups = || {
        vec![
            (1.0f64, sample(6.0, 1.5, 120)),
            (2.0f64, sample(3.5, 1.0, 120)),
            (3.0f64, sample(8.0, 2.0, 120)),
        ]
    };

    for (area, title, with_box) in [
        (&left, "Violins", false),
        (&right, "Violins + embedded box", true),
    ] {
        let mut chart = ChartBuilder::on(area)
            .caption(title, ("sans-serif", 22))
            .margin(20)
            .set_label_area_size(LabelAreaPosition::Left, 45)
            .set_label_area_size(LabelAreaPosition::Bottom, 40)
            .build_cartesian_2d(0.5f64..3.5f64, 0f64..14f64)?;
        chart
            .configure_mesh()
            .x_desc("group")
            .y_desc("value")
            .draw()?;
        chart.draw_series(
            ViolinPlotSeries::from_samples(groups())?
                .width(90)
                .show_box(with_box),
        )?;
    }

    root.present()?;
    println!("wrote violin_plot.svg");
    Ok(())
}