Pandas Groupy chỉ lấy N Nhóm đầu tiên

Question

Tôi có một số DataFrame mà tôi muốn nhóm theo ID, e. G .:

import pandas as pd 
df = pd.DataFrame({'item_id': ['a', 'a', 'b', 'b', 'b', 'c', 'd'], 'user_id': [1,2,1,1,3,1,5]}) 
print df 

nào tạo:

item_id user_id 
0  a  1 
1  a  2 
2  b  1 
3  b  1 
4  b  3 
5  c  1 
6  d  5 

[7 rows x 2 columns] 

tôi có thể dễ dàng nhóm bởi id:

grouped = df.groupby("item_id") 

Nhưng làm thế nào tôi có thể trở lại chỉ có nhóm bằng cách đầu tiên N đối tượng? Ví dụ. Tôi chỉ muốn có 3 item_id duy nhất đầu tiên.

Answer 1

Dưới đây là một cách sử dụng list(grouped).

result = [g[1] for g in list(grouped)[:3]] 

# 1st 
result[0] 

    item_id user_id 
0  a  1 
1  a  2 

# 2nd 
result[1] 

    item_id user_id 
2  b  1 
3  b  1 
4  b  3 

Answer 2

Một phương pháp là sử dụng Counter để có được 3 vật phẩm độc đáo hàng đầu từ danh sách, lọc DataFrame của bạn dựa trên những mặt hàng, và sau đó thực hiện một hoạt động groupby trên DataFrame lọc này.

from collections import Counter 

c = Counter(df.item_id) 
most_common = [item for item, _ in c.most_common(3)] 

>>> df[df.item_id.isin(most_common)].groupby('item_id').sum() 
     user_id 
item_id   
a    3 
b    5 
c    1