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
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85

// This example requires one of the export features to be enabled:
// cargo run --example export_image --features export-default
// cargo run --example export_image --features export-chrome
// cargo run --example export_image --features export-firefox

#[cfg(any(
    feature = "export-chrome",
    feature = "export-firefox",
    feature = "export-default"
))]
use plotlars::{Axis, BoxPlot, CsvReader, Legend, Orientation, Plot, Rgb, Text};

#[cfg(any(
    feature = "export-chrome",
    feature = "export-firefox",
    feature = "export-default"
))]
use plotlars::polars::prelude::*;

#[cfg(any(
    feature = "export-chrome",
    feature = "export-firefox",
    feature = "export-default"
))]
fn main() -> Result<(), Box<dyn std::error::Error>> {
    // Load penguins dataset
    let dataset = CsvReader::new("data/penguins.csv")
        .finish()?
        .lazy()
        .select([
            col("species"),
            col("sex").alias("gender"),
            col("flipper_length_mm").cast(DataType::Int16),
            col("body_mass_g").cast(DataType::Int16),
        ])
        .collect()?;

    // Create a box plot
    let plot = BoxPlot::builder()
        .data(&dataset)
        .labels("species")
        .values("body_mass_g")
        .orientation(Orientation::Vertical)
        .group("gender")
        .box_points(true)
        .point_offset(-1.5)
        .jitter(0.01)
        .opacity(0.1)
        .colors(vec![Rgb(0, 191, 255), Rgb(57, 255, 20), Rgb(255, 105, 180)])
        .plot_title(
            Text::from("Box Plot - Export Example")
                .font("Arial")
                .size(18),
        )
        .x_title(Text::from("species").font("Arial").size(15))
        .y_title(Text::from("body mass (g)").font("Arial").size(15).x(-0.07))
        .legend_title(Text::from("gender").font("Arial").size(15))
        .y_axis(&Axis::new().value_thousands(true))
        .legend(&Legend::new().border_width(1).x(0.9))
        .build();

    // Export in different formats
    println!("Exporting PNG...");
    plot.write_image("output_boxplot.png", 1200, 800, 2.0)?;

    println!("Exporting JPEG...");
    plot.write_image("output_boxplot.jpg", 1200, 800, 1.0)?;

    println!("Exporting SVG...");
    plot.write_image("output_boxplot.svg", 1200, 800, 1.0)?;

    println!("All images exported successfully!");

    Ok(())
}

#[cfg(not(any(
    feature = "export-chrome",
    feature = "export-firefox",
    feature = "export-default"
)))]
fn main() {
    eprintln!("This example requires one of the export features to be enabled.");
    eprintln!("Run with: cargo run --example export_image --features export-default");
}