Skip to content

Compute the quantile per group

Description

Compute the quantile per group

Usage

<LazyGroupBy>$quantile(
  quantile,
  interpolation = c("nearest", "higher", "lower", "midpoint", "linear")
)

Arguments

quantile Quantile between 0.0 and 1.0.
interpolation Interpolation method.

Value

A polars LazyFrame

Examples

library("polars")

lf <- pl$LazyFrame(
  grp = c("c", "c", "a", "c", "a", "b"),
  x = c(0.5, 0.5, 4, 10, 13, 14),
  y = 1:6,
  z = c(TRUE, TRUE, FALSE, TRUE, FALSE, TRUE)
)
lf$collect()

#> shape: (6, 4)
#> ┌─────┬──────┬─────┬───────┐
#> │ grp ┆ x    ┆ y   ┆ z     │
#> │ --- ┆ ---  ┆ --- ┆ ---   │
#> │ str ┆ f64  ┆ i32 ┆ bool  │
#> ╞═════╪══════╪═════╪═══════╡
#> │ c   ┆ 0.5  ┆ 1   ┆ true  │
#> │ c   ┆ 0.5  ┆ 2   ┆ true  │
#> │ a   ┆ 4.0  ┆ 3   ┆ false │
#> │ c   ┆ 10.0 ┆ 4   ┆ true  │
#> │ a   ┆ 13.0 ┆ 5   ┆ false │
#> │ b   ┆ 14.0 ┆ 6   ┆ true  │
#> └─────┴──────┴─────┴───────┘

lf$group_by("grp")$quantile(0.5)$collect()

#> shape: (3, 4)
#> ┌─────┬──────┬─────┬──────┐
#> │ grp ┆ x    ┆ y   ┆ z    │
#> │ --- ┆ ---  ┆ --- ┆ ---  │
#> │ str ┆ f64  ┆ f64 ┆ bool │
#> ╞═════╪══════╪═════╪══════╡
#> │ a   ┆ 13.0 ┆ 5.0 ┆ null │
#> │ c   ┆ 0.5  ┆ 2.0 ┆ null │
#> │ b   ┆ 14.0 ┆ 6.0 ┆ null │
#> └─────┴──────┴─────┴──────┘