How to select rows between a certain date range in python-polars?

Question

If a DataFrame is constructed like the following using polars-python:

import polars as pl
from polars import col
from datetime import datetime

df = pl.DataFrame({
    "dates": ["2016-07-02", "2016-08-10",  "2016-08-31", "2016-09-10"],
    "values": [1, 2, 3, 4]
})

How to select the rows between a certain date range, i.e. between between "2016-08-10" and "2016-08-31", so that the desired outcome is:

┌────────────┬────────┐
│ dates      ┆ values │
│ ---        ┆ ---    │
│ date       ┆ i64    │
╞════════════╪════════╡
│ 2016-08-10 ┆ 2      │
│ 2016-08-31 ┆ 3      │
└────────────┴────────┘

jqurious · Accepted Answer · 2024-09-19 19:40:58Z

9

First you need transform the string values to date types then filter:

# eager
(df.with_columns(pl.col("dates").str.to_date()) 
 .filter(col("dates").is_between(datetime(2016, 8, 9), datetime(2016, 9, 1)))
)

# lazy
(df.lazy()
 .with_columns(pl.col("dates").str.to_date()) 
 .filter(col("dates").is_between(datetime(2016, 8, 9), datetime(2016, 9, 1)))
 .collect()
)

both result in the desired output:

┌────────────┬────────┐
│ dates      ┆ values │
│ ---        ┆ ---    │
│ date       ┆ i64    │
╞════════════╪════════╡
│ 2016-08-10 ┆ 2      │
│ 2016-08-31 ┆ 3      │
└────────────┴────────┘

edited Sep 19, 2024 at 19:40

jqurious

24.2k6 gold badges24 silver badges43 bronze badges

answered Oct 15, 2021 at 22:35

pythonic833

3,2341 gold badge16 silver badges29 bronze badges

Sign up to request clarification or add additional context in comments.

Comments

YouCanDo · Accepted Answer · 2023-02-13 16:33:34Z

0

Use the is_between expression on your dates column:

    df.filter(
            pl.col("dates").is_between(pl.date(2016, 8, 10), pl.date(2016, 8, 31)),
    )

answered Feb 13, 2023 at 16:33

YouCanDo

1

Collectives™ on Stack Overflow

How to select rows between a certain date range in python-polars?

2 Answers 2

Comments

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Related