Nhóm gấu trúc: nhận kích thước của một nhóm biết id của nó (từ .grouper.group_info [0])

Question

Trong đoạn mã sau data là pandas.DataFrame và indices là một tập hợp các cột của data. Sau khi nhóm dữ liệu với groupby Tôi quan tâm đến các id của các nhóm, nhưng chỉ những người có kích thước lớn hơn một ngưỡng (nói: 3).

group_ids=data.groupby(list(data.columns[list(indices)])).grouper.group_info[0] 

Bây giờ, làm cách nào để tìm nhóm nào có kích thước lớn hơn hoặc bằng 3 biết id của nhóm? Tôi chỉ muốn id của các nhóm với một kích thước nhất định.

#TODO: filter out ids from group_ids which correspond to groups with sizes < 3 

Answer 1

Một cách là sử dụng phương pháp size của groupby:

g = data.groupby(...) 
size = g.size() 
size[size > 3] 

Ví dụ, ở đây chỉ có một nhóm các kích thước> 1:

In [11]: df = pd.DataFrame([[1, 2], [3, 4], [1,6]], columns=['A', 'B']) 

In [12]: df 
Out[12]: 
    A B 
0 1 2 
1 3 4 
2 1 6 

In [13]: g = df.groupby('A') 

In [14]: size = g.size() 

In [15]: size[size > 1] 
Out[15]: 
A 
1 2 
dtype: int64 

Nếu bạn đã quan tâm đến việc chỉ hạn chế DataFrame đối với những nhóm lớn, bạn có thể sử dụng phương thức filter:

In [21]: g.filter(lambda x: len(x) > 1) 
Out[21]: 
    A B 
0 1 2 
2 1 6