使用group by的结果筛选数据帧

Question

我似乎找不到解决这个问题的方法，所以我想我会问，因为我可以想象这是一个常见的问题。我有以下数据帧df：

    A   B       Value
0   1   Text 1  Additional comments
1   1   Text 2  TRUE
2   1   Text 3  Apples
3   2   Text 4  Face
4   3   Text 4  Believe
5   4   Text 2  Meat
6   4   Text 9  Chicken
7   5   Text 10 Dairy
8   5   Text 14 Poultry

我想通过在列A和B上分组并消除所有计数小于2的行来过滤这个，这样数据帧的最终结果将是删除第3行和第4行，如下所示

    A   B       Value
0   1   Text 1  Additional comments
1   1   Text 2  TRUE
2   1   Text 3  Apples
5   4   Text 2  Meat
6   4   Text 9  Chicken
7   5   Text 10 Dairy
8   5   Text 14 Poultry

我知道如何通过计数获取组并对其进行过滤：dfg = df.groupby(['A','B']).counts()和dfg = df.groupby(['A','B']).filter(lambda x: len(x) > 1)但是我如何获取这个结果数据帧（dfg）并将其应用到原始数据帧中，以便获得结果。我对数据帧比较陌生，所以这可能是显而易见的，但我已经关注这一点有一段时间了

非常感谢您的帮助！ 非常感谢。 乔恩

Answer 1

groupby和transform，然后比较：

df[df.groupby('A')['A'].transform('size').ge(2)]

---

   A                B                Value
0  1           Text 1  Additional comments
1  1           Text 2                 TRUE
2  1           Text 3               Apples
5  4           Text 2                 Meat
6  4           Text 9              Chicken
7  5           Text 10               Dairy
8  5           Text 14             Poultry