如何将此代码更改为极坐标？“ TypeError：'GroupBy'对象不可下标”

Question

这段代码是pandas。

pandas_reserve_tb \
        .groupby(['hotel_id', 'people_num'])['total_price'] \
        .sum().reset_index()

我想将此代码更改为极坐标。

polars_researve_tb \
        .groupby("hotel_id", "people_num")['total_price'] \
        .sum().with_row_count()

但是，我得到了错误

“类型错误：‘GroupBy’对象不可下标”

如何解决这个错误？

Answer 1

你可能的意思是

polars_researve_tb \
        .groupby(["hotel_id", "people_num"]).agg(pl.col('total_price').sum())

我建议将来发布可重现的示例