在熊猫中选择不同的值groupby列

Question

我有以下代码df1 = df.groupby(['ID_Customer', 'ID_product']).size() 计算每个客户的每种产品的行数。数据集中每个客户的每种产品只有一行。结果是以下df1（的一部分）

    ID cust    ID prod   
    026        009               30
    027        009               1
    028        009               15
    030        009               30
    032        009               30
    ...

如何计算每个客户的独特产品数量？或如何实现按组分组选择不同的值

Answer 1

请尝试以下代码：

df.groupby('ID_Customer')['ID_product'].count()

让我知道这是否对您有用。

谢谢

Answer 2

您可以简单地使用nunique：

df.groupby(['ID_Customer'])['ID_product'].nunique()