plotlars 0.12.0

Plotlars is a Rust library designed to facilitate the integration between the Polars data analysis library and visualization libraries.
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
56
57
58
59
60
61

use plotlars::polars::prelude::*;
use plotlars::{ColorBar, Palette, Plot, SurfacePlot, Text};
use std::iter;

fn main() {
    let n: usize = 100;
    let x_base: Vec<f64> = (0..n)
        .map(|i| {
            let step = (10.0 - (-10.0)) / (n - 1) as f64;
            -10.0 + step * i as f64
        })
        .collect();
    let y_base: Vec<f64> = (0..n)
        .map(|i| {
            let step = (10.0 - (-10.0)) / (n - 1) as f64;
            -10.0 + step * i as f64
        })
        .collect();

    let x = x_base
        .iter()
        .flat_map(|&xi| iter::repeat_n(xi, n))
        .collect::<Vec<_>>();

    let y = y_base
        .iter()
        .cycle()
        .take(n * n)
        .cloned()
        .collect::<Vec<_>>();

    let z = x_base
        .iter()
        .flat_map(|i| {
            y_base
                .iter()
                .map(|j| 1.0 / (j * j + 5.0) * j.sin() + 1.0 / (i * i + 5.0) * i.cos())
                .collect::<Vec<_>>()
        })
        .collect::<Vec<_>>();

    let dataset = df![
        "x" => &x,
        "y" => &y,
        "z" => &z,
    ]
    .unwrap();

    SurfacePlot::builder()
        .data(&dataset)
        .x("x")
        .y("y")
        .z("z")
        .plot_title(Text::from("Surface Plot").font("Arial").size(18))
        .color_bar(&ColorBar::new().border_width(1))
        .color_scale(Palette::Cividis)
        .reverse_scale(true)
        .opacity(0.5)
        .build()
        .plot();
}