scatterplot/
scatterplot.rs

1use polars::prelude::*;
2
3use plotlars::{Axis, Legend, Plot, Rgb, ScatterPlot, Shape, Text, TickDirection};
4
5fn main() {
6    let dataset = LazyCsvReader::new(PlPath::new("data/penguins.csv"))
7        .finish()
8        .unwrap()
9        .select([
10            col("species"),
11            col("sex").alias("gender"),
12            col("flipper_length_mm").cast(DataType::Int16),
13            col("body_mass_g").cast(DataType::Int16),
14        ])
15        .collect()
16        .unwrap();
17
18    let axis = Axis::new()
19        .show_line(true)
20        .tick_direction(TickDirection::OutSide)
21        .value_thousands(true);
22
23    ScatterPlot::builder()
24        .data(&dataset)
25        .x("body_mass_g")
26        .y("flipper_length_mm")
27        .group("species")
28        .opacity(0.5)
29        .size(12)
30        .colors(vec![Rgb(178, 34, 34), Rgb(65, 105, 225), Rgb(255, 140, 0)])
31        .shapes(vec![Shape::Circle, Shape::Square, Shape::Diamond])
32        .plot_title(Text::from("Scatter Plot").font("Arial").size(20).x(0.065))
33        .x_title("body mass (g)")
34        .y_title("flipper length (mm)")
35        .legend_title("species")
36        .x_axis(&axis.clone().value_range(vec![2500.0, 6500.0]))
37        .y_axis(&axis.clone().value_range(vec![170.0, 240.0]))
38        .legend(&Legend::new().x(0.85).y(0.15))
39        .build()
40        .plot();
41}
scatterplot/scatterplot.rs

scatterplot/
scatterplot.rs
