polars-qt 0.1.1

No project description provided

polars_qt

Add a qt namespace for polars.

Currently :

• a rolling_rank function

• A if-then expression.

rolling_rank

import polars as pl
import polars_qt as pq
df = pl.DataFrame({
    'a': [5.2, 4.1, 6.3, None, 10, 4, 5],
})
df.with_columns(
    pq.rolling_rank(pl.col('a'), 4, min_periods=1, pct=True).alias('a_rank'),
    pl.col('a').qt.rolling_rank(4, pct=False, rev=True).alias('a_rank2')
)

shape: (7, 3)
┌──────┬──────────┬─────────┐
│ a    ┆ a_rank   ┆ a_rank2 │
│ ---  ┆ ---      ┆ ---     │
│ f64  ┆ f64      ┆ f64     │
╞══════╪══════════╪═════════╡
│ 5.2  ┆ 1.0      ┆ null    │
│ 4.1  ┆ 0.5      ┆ 2.0     │
│ 6.3  ┆ 1.0      ┆ 1.0     │
│ null ┆ null     ┆ null    │
│ 10.0 ┆ 1.0      ┆ 1.0     │
│ 4.0  ┆ 0.333333 ┆ 3.0     │
│ 5.0  ┆ 0.666667 ┆ 2.0     │
└──────┴──────────┴─────────┘

If-then

df = pl.DataFrame({
    'g': ['a', 'a', 'b', 'a', 'b'],
    'v': [1, 3, 5, 2, 4],
})
df.select(pl.col('v').qt.if_then((pl.len()>2), pl.col('v')*2).over('g'))

shape: (5, 1)
┌─────┐
│ v   │
│ --- │
│ i64 │
╞═════╡
│ 2   │
│ 6   │
│ 5   │
│ 4   │
│ 4   │
└─────┘