Polars 计算当年每月各有多少天

说明

《Python Polars》教程正在筹划更新中，敬请期待！Polars 是一个高性能的数据处理库，特别适用于大数据集和需要并行计算的场景。如果你在处理大数据时遇到性能瓶颈，可以考虑尝试 Polars，它的延迟计算和并行化机制可以显著提升处理效率。

如果你记不住每个月有多少天，可以使用以下 Polars 代码，计算出指定年份每月有多少天，供你来查询。

代码如下：

import polars as pl

print(
    pl.date_range(
        start=pl.date(2025, 1, 1),
        end=pl.date(2026, 1, 1), 
        interval='1d',
        closed='left', 
        eager=True
    )
    .to_frame('d')
    .group_by_dynamic('d', every='1mo')
    .agg(
        pl.col('d').dt.month().first().alias('month'),
        pl.len().alias('days_in_month'),
    )
    .drop('d')
)

输出：

shape: (12, 2)
┌───────┬───────────────┐
│ month ┆ days_in_month │
│ ---   ┆ ---           │
│ i8    ┆ u32           │
╞═══════╪═══════════════╡
│ 1     ┆ 31            │
│ 2     ┆ 28            │
│ 3     ┆ 31            │
│ 4     ┆ 30            │
│ 5     ┆ 31            │
│ …     ┆ …             │
│ 8     ┆ 31            │
│ 9     ┆ 30            │
│ 10    ┆ 31            │
│ 11    ┆ 30            │
│ 12    ┆ 31            │
└───────┴───────────────┘

用 Polars 构造全年日期的时间序列，再按月分组数据，聚合计算得到结果。

（完）

Polars 计算当年每月各有多少天

相关内容