python-3.x - 在 Pandas 中选择最大值特殊条件？

Question

我有一个像

customer_id    product_code    No_visits
1              90              15
1              105             30
1              100             45
2              50              60
2              100             45
2              10              30

我要选择no_visits最高的customer_id、product_code对应的

customer_id    product_code    No_visits
1              100             45
2              50              60

df.groupby('customer_id')['product_code, no_visits'].max()

这给了我最大 product_code 和最大 no_visits 的结果，这绝对不是我想要的。请帮忙

Answer 1

您可以使用

>>> df[df.No_visits == df.No_visits.groupby(df.customer_id).transform(max)]

   customer_id  product_code  No_visits
2            1           100         45
3            2            50         60

说明

df.No_visits == df.No_visits.groupby(df.customer_id).transform(max)

对于每一行，查找该行的客户 ID 的最大访问次数。然后，您只需过滤那些访问次数等于该数字的行。