我正试着用北极熊取代一些Pandas .我是北极星的新手.
原始代码:
return pd.DataFrame(
data.groupby([list_of_fields]).size(),
columns=["Count"],
).reset_index()
我被困在如何命名列"计数".到目前为止，我已经try 过:
return pl.DataFrame(data.groupby([list_of_fields]).n_unique( "Count"))
但这给了我们:
TypeError: n_unique() takes 1 positional argument but 2 were given
n_unique似乎仅在未附加到groupby时才允许使用名称参数.我怎么能这样做呢？
在Pandas 中，DataFrameGroupBy.size是计算每组的行数.
在极性中，这可以通过使用pl.len来实现.我们从一些示例数据开始.
import polars as pl
df = pl.DataFrame({
"group": ["A", "A", "A", "B", "B", "C"],
})
shape: (6, 1)
┌───────┐
│ group │
│ --- │
│ str │
╞═══════╡
│ A │
│ A │
│ A │
│ B │
│ B │
│ C │
└───────┘
现在，使用
df.group_by("group", maintain_order=True).agg(pl.len().alias("Count"))
赠送
shape: (3, 2)
┌───────┬───────┐
│ group ┆ Count │
│ --- ┆ --- │
│ str ┆ u32 │
╞═══════╪═══════╡
│ A ┆ 3 │
│ B ┆ 2 │
│ C ┆ 1 │
└───────┴───────┘
请注意，我们已经使用pl.Expr.alias重命名了group_by上下文中的pl.len()表达式.