Python-Polars 中的 GroupBy、列选择和平均值

Question

我正在尝试翻译这个 pandas 声明：

  df.groupby(['X', 'Y'])['Z'].mean()

到极地，但我被困在这一点上：

  df.group_by(['X', 'Y'], maintain_order=True).select(pl.col('Z').mean())

我在哪里收到此错误：

  AttributeError: 'GroupBy' object has no attribute 'select'

任何帮助将不胜感激！

Answer 1

你很接近

df.group_by(['X', 'Y'], maintain_order=True).agg(pl.col('Z').mean())