Pandas groupby 然后删除低于指定大小的组 [英] Pandas groupby then drop groups below specified size

问题描述

我正在尝试将 DataFrame 分成多个组并将组删除到最小尺寸以下(小的异常值).

I'm trying to separate a DataFrame into groups and drop groups below a minimum size (small outliers).

这是我尝试过的:

df.groupby(['A']).filter(lambda x: x.count() > min_size)
df.groupby(['A']).filter(lambda x: x.size() > min_size)
df.groupby(['A']).filter(lambda x: x['A'].count() > min_size)
df.groupby(['A']).filter(lambda x: x['A'].size() > min_size)

但是这些要么抛出异常，要么返回与我预期不同的表.我只想过滤，而不是计算新表.

But these either throw an exception or return a different table than I'm expecting. I'd just like to filter, not compute a new table.

推荐答案

你可以使用len:

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

In [12]: df.groupby('A').filter(lambda x: len(x) > 1)
Out[12]:
   A  B
0  1  2
1  1  4

这篇关于Pandas groupby 然后删除低于指定大小的组的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！