使用熊貓數據框如何將計數應用於多級分組列？

Question

我有一個數據框與多個列，我想使用數組後，這樣它應用於2列或更多列的組合。例如，假設我有兩列：

user_id product_name 
1  Apple 
1  Banana 
1  Apple 
2  Carrot 
2  Tomato 
2  Carrot 
2  Tomato 
3  Milk 
3  Cucumber 

...

我想實現的是這樣的：

user_id product_name Product_Count_per_User 
1  Apple  1 
1  Banana  2 
2  Carrot  2 
2  Tomato  2 
3  Milk   1 
3  Cucumber  1 

我無法得到它。我嘗試這樣做：

dcf6 = df3.groupby(['user_id','product_name'])['user_id', 'product_name'].count() 

，但似乎並沒有得到我想要的東西，它會顯示4列，而不是3。如何做呢？謝謝。

Answer 1

您在同一時間計數兩列，你可以使用groupby.size：

(df.groupby(['user_id', 'Product_Name']).size() 
    .rename('Product_Count_per_User').reset_index()) 

或計數只有一列：

df.groupby(['user_id','Product_Name'])['user_id'].size() 

Answer 2

使用GroupBy.size：

dcf6 = df3.groupby(['user_id','Product_Name']).size() 
      .reset_index(name='Product_Count_per_User') 
print (dcf6) 
    user_id Product_Name Product_Count_per_User 
0  1  Apple      2 
1  1  Banana      1 
2  2  Carrot      2 
3  2  Tomato      2 
4  3  Cucumber      1 
5  3   Milk      1 

What is the difference between size and count in pandas?

Answer 3

根據您自己的代碼，只需執行此操作。

df.groupby(['user_id','product_name'])['user_id']. 
      agg({'Product_Count_per_User':'count'}).reset_index(level=1) 

     product_name Product_Count_per_User 
user_id      
1    Apple  2 
1    Banana  1 
2    Carrot  2 
2    Tomato  2 
3   Cucumber  1 
3    Milk  1