集團由兩列和計算每個組合的大熊貓出現

Question

，我有以下的數據幀：

data = pd.DataFrame({'user_id' : ['a1', 'a1', 'a1', 'a2','a2','a2','a3','a3','a3'], 'product_id' : ['p1','p1','p2','p1','p1','p1','p2','p2','p3']}) 

product_id user_id 
    p1  a1 
    p1  a1 
    p2  a1 
    p1  a2 
    p1  a2 
    p1  a2 
    p2  a3 
    p2  a3 
    p3  a3 

在現實情況下，可能有一些其他列一樣好，但我需要做的是組由數據幀由PRODUCT_ID和USER_ID列和計數每個組合的數量和在一個新的DAT幀添加作爲新的列

輸出應該是這樣的：

user_id product_id count 
a1  p1   2 
a1  p2   1 
a2  p1   3 
a3  p2   2 
a3  p3   1 

我曾嘗試下面的代碼：

grouped=data.groupby(['user_id','product_id']).count() 

但結果是：

user_id product_id 
a1  p1 
      p2 
a2  p1 
a3  p2 
      p3 

其實對我來說最重要的是有一個具有發生次數列名計數，我需要稍後使用該列。

Answer 1

也許這就是你想要的？

>>> data = pd.DataFrame({'user_id' : ['a1', 'a1', 'a1', 'a2','a2','a2','a3','a3','a3'], 'product_id' : ['p1','p1','p2','p1','p1','p1','p2','p2','p3']}) 
>>> count_series = data.groupby(['user_id', 'product_id']).size() 
>>> count_series 
user_id product_id 
a1  p1   2 
     p2   1 
a2  p1   3 
a3  p2   2 
     p3   1 
dtype: int64 
>>> new_df = count_series.to_frame(name = 'size').reset_index() 
>>> new_df 
    user_id product_id size 
0  a1   p1  2 
1  a1   p2  1 
2  a2   p1  3 
3  a3   p2  2 
4  a3   p3  1 
>>> new_df['size'] 
0 2 
1 1 
2 3 
3 2 
4 1 
Name: size, dtype: int64