python - Pandas：获取某个类别中最受欢迎的项目 ID（两个 Groupby 级别）

Question

对不起，我想不出一个标题。

我有一个这样的DF：

item_cat1	item_id	item_count
一个	1	2
乙	2	5
一个	3	4
C	4	2

我想获得 item category 中最受欢迎的 item item_cat1。这是所需的输出：

item_cat1	item_id	item_count
一个	3	4
乙	2	5
C	4	2

说明：在 item_cat1 A 中，3与 item_id (2) 相比，item_id 的销量最高 (4 1)。

我试过train.groupby(["item_cat1", "item_id"])["item_count"].sum()了，但我不知道如何只选择最大值。

P/S：我想要的是item_id，而不是item_count。这个答案对我没有帮助：Get the row(s) which have the max value in groups using groupby

标签： pythonpandas