如何在Polars中按分组获取行序号row_count

使用Polars处理数据时，若需要按指定分组为每行生成对应的组内行序号，可参考如下实现方式：

示例代码 #

out_df = df.with_columns(
    pl.col("md5").row_count().over("md5").alias("row_count"),
)

效果演示 #

处理前的测试数据：

df = pl.DataFrame({"md5": ["a", "a", "b"]})

处理后的输出结果：

shape: (3, 2)
┌─────┬───────────┐
│ md5 ┆ row_count │
│ --- ┆ ---       │
│ str ┆ i32       │
╞═════╪═══════════╡
│ a   ┆ 1         │
│ a   ┆ 2         │
│ b   ┆ 1         │
└─────┴───────────┘

内容的提问来源于stack exchange，提问作者侯颖堃