python - 按组滚动计数独特的分类特征

Question

我想按组获得唯一分类的滚动计数:

Group  Item
A      pen
A      pen
A      elbow
A      warthog
B      elbow
B      peach

应该导致:

Group  Item     Unique_item_count
A      pen      1
A      pen      1
A      elbow    2
A      warthog  3
B      elbow    1
B      peach    2

我觉得 pd.rolling_count 可能有答案，但我还没弄明白。感谢您的智慧和魔法!

Answer 1

我们可以 GroupBy 两次。首先我们得到 nunique 值，第二次我们得到 cumsum 使我们对 Item 中每个唯一值的计数增加:

然后我们将这些结果合并回我们的原始数据框。

s = df.groupby(['Group', 'Item'], sort=False)['Item'].nunique().groupby(level=0).cumsum()

final = df.merge(s.reset_index(name='Unique_item_count'), on=['Group', 'Item'])

输出

  Group     Item  Unique_item_count
0     A      pen                  1
1     A      pen                  1
2     A    elbow                  2
3     A  warthog                  3
4     B    elbow                  1
5     B    peach                  2