如何在Pandas groupby中获得独特计数

Question

我想得到每个订单数量的不同产品。我设法得到了总的_产品计数（感谢另一个SO用户的帮助），但我无法计算出不同的计数。在

这就是我所拥有的：

data['total_productcount'] = data.groupby(['order_number'])['order_number'].transform('size')

它给出了：

^{pr2}$

这是我要生成的数据帧（包括：distict_productcount）

order_number          product_id      total_productcount   distict_productcount
171-1046037-0511522   4260179734731   5                    1
171-1046037-0511522   4054673034394   5                    1
171-1046037-0511522   4054673001235   5                    1
171-1046037-0511522   4054673005752   5                    1
171-1046037-0511522   5011385960075   5                    1
171-1046037-0511522   5011385960075   5                    2

如何生成“distict_productcount”？在

Answer 1

data.groupby('order_number').product_id.nunique()

您可以使用transform或join来获取新列

通过transform

^{pr2}$

通过join

s = data.groupby('order_number').product_id.nunique()
df = df.join(s.rename('distinct_productcount'), on='order_number')