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

use plotlars::polars::prelude::*;
use plotlars::{Axis, CsvReader, Legend, Line, Plot, Rgb, Shape, Text, TimeSeriesPlot};

fn main() {
    let revenue_dataset = CsvReader::new("data/revenue_and_cost.csv")
        .finish()
        .unwrap()
        .lazy()
        .select([
            col("Date").cast(DataType::String),
            col("Revenue").cast(DataType::Int32),
            col("Cost").cast(DataType::Int32),
        ])
        .collect()
        .unwrap();

    TimeSeriesPlot::builder()
        .data(&revenue_dataset)
        .x("Date")
        .y("Revenue")
        .additional_series(vec!["Cost"])
        .size(8)
        .colors(vec![Rgb(0, 0, 255), Rgb(255, 0, 0)])
        .lines(vec![Line::Dash, Line::Solid])
        .shapes(vec![Shape::Circle, Shape::Square])
        .plot_title(Text::from("Time Series Plot").font("Arial").size(18))
        .legend(&Legend::new().x(0.05).y(0.9))
        .x_title("x")
        .y_title(Text::from("y").color(Rgb(0, 0, 255)))
        .y2_title(Text::from("y2").color(Rgb(255, 0, 0)))
        .y_axis(
            &Axis::new()
                .value_color(Rgb(0, 0, 255))
                .show_grid(false)
                .zero_line_color(Rgb(0, 0, 0)),
        )
        .y2_axis(
            &Axis::new()
                .axis_side(plotlars::AxisSide::Right)
                .value_color(Rgb(255, 0, 0))
                .show_grid(false),
        )
        .build()
        .plot();

    let temperature_dataset = CsvReader::new("data/debilt_2023_temps.csv")
        .has_header(true)
        .try_parse_dates(true)
        .finish()
        .unwrap()
        .lazy()
        .with_columns(vec![
            (col("tavg") / lit(10)).alias("tavg"),
            (col("tmin") / lit(10)).alias("tmin"),
            (col("tmax") / lit(10)).alias("tmax"),
        ])
        .collect()
        .unwrap();

    TimeSeriesPlot::builder()
        .data(&temperature_dataset)
        .x("date")
        .y("tavg")
        .additional_series(vec!["tmin", "tmax"])
        .colors(vec![Rgb(128, 128, 128), Rgb(0, 122, 255), Rgb(255, 128, 0)])
        .lines(vec![Line::Solid, Line::Dot, Line::Dot])
        .plot_title("Temperature at De Bilt (2023)")
        .legend_title("Legend")
        .build()
        .plot();
}